[https://nvbugs/5474119][fix] Re-enable test (NVIDIA#8809)

dongfengy · fredricz-20070104 · commit 7765957c3ec1 · 2025-11-05T10:51:43.000+08:00
Signed-off-by: Dongfeng Yu &lt;dongfengy@nvidia.com&gt;
Signed-off-by: FredricZ-2007 &lt;226039983+fredricz-20070104@users.noreply.github.com&gt;
diff --git 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(

Original file line number	Diff line number	Diff line change
`@@ -1840,7 +1840,7 @@ def test_moe_fp8_per_tensor_scale(num_tokens, expert_info, hidden_size,`
`1840`	`1840`	`reason="The kernel only supports Blackwell. Current SM is %d." %`
`1841`	`1841`	`getSMVersion(),`
`1842`	`1842`	`)`
`1843`		`-@pytest.mark.parametrize("num_tokens", [1, 1024])`
	`1843`	`+@pytest.mark.parametrize("num_tokens", [1, 256, 1024])`
`1844`	`1844`	`@pytest.mark.parametrize("hidden_size", [512])`
`1845`	`1845`	`@pytest.mark.parametrize("intermediate_size", [512])`
`1846`	`1846`	`@pytest.mark.parametrize(`