From ffab217974a640b1a199f6a76de8841130f7a7f7 Mon Sep 17 00:00:00 2001 From: Enwei Zhu <21126786+syuoni@users.noreply.github.com> Date: Mon, 26 Jan 2026 08:34:17 +0800 Subject: [PATCH] [None][fix] Fix CuteDSL MoE unittest (#10983) Signed-off-by: Enwei Zhu <21126786+syuoni@users.noreply.github.com> --- tests/unittest/_torch/thop/parallel/test_cute_dsl_moe.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/unittest/_torch/thop/parallel/test_cute_dsl_moe.py b/tests/unittest/_torch/thop/parallel/test_cute_dsl_moe.py index b176c03685..91024f5e4b 100644 --- a/tests/unittest/_torch/thop/parallel/test_cute_dsl_moe.py +++ b/tests/unittest/_torch/thop/parallel/test_cute_dsl_moe.py @@ -740,6 +740,8 @@ def test_nvfp4_gather_grouped_gemm_swiglu_blackwell( token_final_scales, token_selected_experts = routing_logits.topk(top_k, dim=-1) token_selected_experts = token_selected_experts.to(torch.int32) token_final_scales = token_final_scales.softmax(dim=-1).to(torch.float32) + # Ensure at least one valid token + token_selected_experts[0] = 0 ( tile_idx_to_group_idx,