From 52b8384d1088551c1829358109a1ccf06fa626ab Mon Sep 17 00:00:00 2001 From: Cheng Date: Wed, 10 Sep 2025 14:22:22 +0900 Subject: [PATCH] Fix flaky addmm tests (#2581) --- python/tests/test_blas.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tests/test_blas.py b/python/tests/test_blas.py index 67289ceef..dc9e93699 100644 --- a/python/tests/test_blas.py +++ b/python/tests/test_blas.py @@ -702,7 +702,7 @@ class TestBlas(mlx_tests.MLXTestCase): b = mx.ones((5, 5)) out = mx.addmm(a, b, a, beta=beta, alpha=alpha) expected = beta * a + alpha * (b @ a) - self.assertTrue(mx.allclose(expected, out)) + self.assertTrue(mx.allclose(expected, out, atol=1e-5)) # Broadcast c a = mx.ones((5, 5)) @@ -710,7 +710,7 @@ class TestBlas(mlx_tests.MLXTestCase): c = mx.ones((1, 5)) out = mx.addmm(c, a, b, beta=beta, alpha=alpha) expected = beta * c + alpha * (a @ b) - self.assertTrue(mx.allclose(expected, out)) + self.assertTrue(mx.allclose(expected, out, atol=1e-5)) def test_addmm_grad(self): def make_ref_addmm(alpha, beta):