From 6f81ed72aeaf7b7e575482f5899baef50f2afd5d Mon Sep 17 00:00:00 2001 From: Avshalom Date: Sun, 12 Jan 2025 10:21:54 +0200 Subject: [PATCH] fixing test --- tests/kernels/test_moe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/kernels/test_moe.py b/tests/kernels/test_moe.py index e0db5129340a3..7fa5de1984452 100644 --- 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,