Skip to content

GitLab

Unverified Commit aac0b817 authored by Fadi Arafeh's avatar Fadi Arafeh Committed by GitHub
Browse files

[CPU Backend][BugFix] Fix failing CPU MoE test (#32876) 


Signed-off-by: default avatarFadi Arafeh <fadi.arafeh@arm.com>
parent 05f3d714
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/kernels/moe/test_moe.py
View file @ aac0b817
...@@ -1311,6 +1311,7 @@ def test_moe_sum(m: int, topk: int, k: int, dtype: torch.dtype): ...@@ -1311,6 +1311,7 @@ def test_moe_sum(m: int, topk: int, k: int, dtype: torch.dtype):
opcheck(torch.ops._moe_C.moe_sum, (input, actual)) opcheck(torch.ops._moe_C.moe_sum, (input, actual))
@pytest.mark.usefixtures("default_vllm_config")
@pytest.mark.parametrize("m", [1, 33]) @pytest.mark.parametrize("m", [1, 33])
@pytest.mark.parametrize("n,k", [(128, 128)]) @pytest.mark.parametrize("n,k", [(128, 128)])
@pytest.mark.parametrize("e", [8]) @pytest.mark.parametrize("e", [8])
......
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