Skip to content

Commit c6c10ca

Browse files
authored
[Bugfix] Reduce moe_sum test size to avoid OOM (vllm-project#18484)
Signed-off-by: Bill Nell <bnell@redhat.com>
1 parent c154d89 commit c6c10ca

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/kernels/moe/test_moe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -577,7 +577,7 @@ def test_moe_align_block_size_opcheck():
577577
num_tokens_post_pad))
578578

579579

580-
@pytest.mark.parametrize("m", [1, 33, 222, 1024 * 128])
580+
@pytest.mark.parametrize("m", [1, 33, 64, 222])
581581
@pytest.mark.parametrize("topk", TOP_KS)
582582
@pytest.mark.parametrize("k", [128, 511, 1024])
583583
@pytest.mark.parametrize("dtype",

0 commit comments

Comments
 (0)