Skip to content

GitLab

Unverified Commit c6c10ca9 authored by bnellnm's avatar bnellnm Committed by GitHub
Browse files

[Bugfix] Reduce moe_sum test size to avoid OOM (#18484) 


Signed-off-by: default avatarBill Nell <bnell@redhat.com>
parent c154d893
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/kernels/moe/test_moe.py
View file @ c6c10ca9
......@@ -577,7 +577,7 @@ def test_moe_align_block_size_opcheck():
num_tokens_post_pad))
@pytest.mark.parametrize("m", [1, 33, 222, 1024 * 128])
@pytest.mark.parametrize("m", [1, 33, 64, 222])
@pytest.mark.parametrize("topk", TOP_KS)
@pytest.mark.parametrize("k", [128, 511, 1024])
@pytest.mark.parametrize("dtype",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment