From 0edba5a7e22c5519276b94c18338aaa920838275 Mon Sep 17 00:00:00 2001 From: dongfengy <99041270+dongfengy@users.noreply.github.com> Date: Fri, 31 Oct 2025 10:17:58 -0700 Subject: [PATCH] [https://nvbugs/5474119][fix] Re-enable test (#8809) Signed-off-by: Dongfeng Yu --- tests/unittest/_torch/thop/parallel/test_moe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unittest/_torch/thop/parallel/test_moe.py b/tests/unittest/_torch/thop/parallel/test_moe.py index 00112202d3..ddfa2eca65 100644 --- a/tests/unittest/_torch/thop/parallel/test_moe.py +++ b/tests/unittest/_torch/thop/parallel/test_moe.py @@ -1840,7 +1840,7 @@ def test_moe_fp8_per_tensor_scale(num_tokens, expert_info, hidden_size, reason="The kernel only supports Blackwell. Current SM is %d." % getSMVersion(), ) -@pytest.mark.parametrize("num_tokens", [1, 1024]) +@pytest.mark.parametrize("num_tokens", [1, 256, 1024]) @pytest.mark.parametrize("hidden_size", [512]) @pytest.mark.parametrize("intermediate_size", [512]) @pytest.mark.parametrize(