Skip to content

Commit 3c06efd

Browse files
ElizaWszolatlrmchlsmth
authored andcommitted
[Misc] Bump up test_fused_moe tolerance (vllm-project#10364)
Signed-off-by: ElizaWszola <eliza@neuralmagic.com> Signed-off-by: Tyler Michael Smith <tyler@neuralmagic.com>
1 parent 346353e commit 3c06efd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/kernels/test_moe.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def test_fused_moe(
4545
score = torch.randn((m, e), device="cuda", dtype=dtype)
4646
triton_output = fused_moe(a, w1, w2, score, topk, renormalize=False)
4747
torch_output = torch_moe(a, w1, w2, score, topk)
48-
torch.testing.assert_close(triton_output, torch_output, atol=1e-2, rtol=0)
48+
torch.testing.assert_close(triton_output, torch_output, atol=2e-2, rtol=0)
4949

5050

5151
@pytest.mark.parametrize("dtype",

0 commit comments

Comments
 (0)