Skip to content

Commit e1f3b2d

Browse files
committed
Remove non-gated case from test
Signed-off-by: Daniel Serebrenik <daserebrenik@nvidia.com>
1 parent 922d36e commit e1f3b2d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/kernels/moe/test_ocp_mx_moe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -988,7 +988,7 @@ def dequant_mxfp4_batches(mat_fp4: torch.Tensor, scale_tensor: torch.Tensor):
988988
@pytest.mark.parametrize("num_experts", [32])
989989
@pytest.mark.parametrize("num_tokens", [1, 128])
990990
@pytest.mark.parametrize("intermediate_size,hidden_size", [(3072, 3072)])
991-
@pytest.mark.parametrize("is_gated", [True, False], ids=["gated", "non_gated"])
991+
@pytest.mark.parametrize("is_gated", [True], ids=["gated"])
992992
@pytest.mark.skipif(
993993
not TRTLLM_GEN_MXFP8_AVAILABLE,
994994
reason="nvidia gpu and compute capability sm100 is required for this test",

0 commit comments

Comments
 (0)