We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 691a3ec commit 79ee45bCopy full SHA for 79ee45b
tests/kernels/test_moe.py
@@ -45,7 +45,7 @@ def test_fused_moe(
45
score = torch.randn((m, e), device="cuda", dtype=dtype)
46
triton_output = fused_moe(a, w1, w2, score, topk, renormalize=False)
47
torch_output = torch_moe(a, w1, w2, score, topk)
48
- torch.testing.assert_close(triton_output, torch_output, atol=1e-2, rtol=0)
+ torch.testing.assert_close(triton_output, torch_output, atol=2e-2, rtol=0)
49
50
51
@pytest.mark.parametrize("dtype",
0 commit comments