xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-13 19:15:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: Bill Nell <bnell@redhat.com>
This commit is contained in:
bnellnm 2025-05-21 09:46:39 -04:00 committed by GitHub
parent c154d89306
commit c6c10ca920
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/kernels/moe/test_moe.py
View File

@ -577,7 +577,7 @@ def test_moe_align_block_size_opcheck():
num_tokens_post_pad)) 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("topk", TOP_KS)
@pytest.mark.parametrize("k", [128, 511, 1024]) @pytest.mark.parametrize("k", [128, 511, 1024])
@pytest.mark.parametrize("dtype", @pytest.mark.parametrize("dtype",