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

[Bugfix] Fix test_mixtral_moe (#24371)

Browse Source
This commit is contained in:
Jee Jee Li 2025-09-07 00:32:03 +08:00 committed by GitHub
parent 00a4e56d8d
commit 7555d6b34a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -371,8 +371,8 @@ def test_fused_moe_wn16(m: int, n: int, k: int, e: int, topk: int,
@pytest.mark.parametrize( @pytest.mark.parametrize(
"use_rocm_aiter", [True, False] if current_platform.is_rocm() else [False]) "use_rocm_aiter", [True, False] if current_platform.is_rocm() else [False])
@torch.inference_mode() @torch.inference_mode()
def test_mixtral_moe(dtype: torch.dtype, padding: bool, use_rocm_aiter: bool, def test_mixtral_moe(dist_init, dtype: torch.dtype, padding: bool,
monkeypatch): use_rocm_aiter: bool, monkeypatch):
"""Make sure our Mixtral MoE implementation agrees with the one from """Make sure our Mixtral MoE implementation agrees with the one from
huggingface.""" huggingface."""