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

Signed-off-by: Fadi Arafeh <fadi.arafeh@arm.com>
2026-01-23 12:06:51 +00:00
parent 05f3d714db
commit aac0b817fa
1 changed files with 1 additions and 0 deletions
--- a/tests/kernels/moe/test_moe.py
+++ b/tests/kernels/moe/test_moe.py
@@ -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))


+@pytest.mark.usefixtures("default_vllm_config")
@pytest.mark.parametrize("m", [1, 33])
@pytest.mark.parametrize("n,k", [(128, 128)])
@pytest.mark.parametrize("e", [8])