[ROCm][CI] Handle tied experts in softplus sqrt top-k test (#52679)
Signed-off-by: Andreas Karatzas <Andreas.Karatzas@amd.com> Signed-off-by: Andreas Karatzas <akaratza@amd.com>
Andreas Karatzas committed
Sep 1, 2026 at 18:04 UTC
e01d4acbb1ea559379825072afd7595fd0160d1f
1 file changed
+2
tests/kernels/moe/test_topk_softplus_sqrt.py
+2
@@ -277,6 +277,8 @@ def test_fused_topk_softplus_sqrt_padding(
277
indices_dtype = torch.int32
278
279
gating_output = torch.randn((num_tokens, num_experts), dtype=dtype, device="cuda")
280
+ # Exercise top-k cutoff ties without relying on random low-precision collisions.
281
+ gating_output[0] = 0
282
283
padding_rows = torch.zeros(num_tokens, dtype=torch.bool, device="cuda")
284
padding_rows[1::2] = True