From f46104deeba201f23cc3c68693d33ba8f346fa42 Mon Sep 17 00:00:00 2001 From: Fardin Hoque Date: Fri, 17 Oct 2025 10:11:52 -0700 Subject: [PATCH] Update tests/kernels/moe/test_batched_moe.py Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com> Signed-off-by: Fardin Hoque --- tests/kernels/moe/test_batched_moe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/kernels/moe/test_batched_moe.py b/tests/kernels/moe/test_batched_moe.py index e607b2ccc93c..abbcc1904455 100644 --- a/tests/kernels/moe/test_batched_moe.py +++ b/tests/kernels/moe/test_batched_moe.py @@ -95,7 +95,7 @@ class BatchedMMTensors: @pytest.mark.parametrize("num_experts", [8, 32]) -@pytest.mark.parametrize("max_tokens_per_expert", [32, 512]) +@pytest.mark.parametrize("max_tokens_per_expert", [32, 224, 512]) @pytest.mark.parametrize("K", [128, 1024]) @pytest.mark.parametrize("N", [128, 1024]) @pytest.mark.parametrize("dtype", [torch.float8_e4m3fn, torch.bfloat16])