Skip to content

Commit 79ee45b

Browse files
authored
[Misc] Bump up test_fused_moe tolerance (#10364)
Signed-off-by: ElizaWszola <[email protected]>
1 parent 691a3ec commit 79ee45b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/kernels/test_moe.py

Lines changed: 1 addition & 1 deletion
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)