fixing test

vllm-project · Jan 12, 2025 · 6f81ed7 · 6f81ed7
1 parent 83b6002
commit 6f81ed7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/kernels/test_moe.py b/tests/kernels/test_moe.py
@@ -48,7 +48,7 @@ def test_fused_moe(
     triton_output = fused_moe(a, w1, w2, score, topk, renormalize=False)
     torch_output = torch_moe(a, w1, w2, score, topk)
     torch.testing.assert_close(triton_output, torch_output, atol=2e-2, rtol=0)
-    iterative_output = iterative_moe(a, w1, w2, score, topk)
+    iterative_output = iterative_moe(a, w1, w2, score, topk, renormalize=False)
     torch.testing.assert_close(iterative_output,
                                torch_output,
                                atol=2e-2,