From 02afae04f4c7e02f5d7eb0ff1b699f81c43c446e Mon Sep 17 00:00:00 2001 From: chenyu Date: Sun, 1 Feb 2026 11:24:58 -0500 Subject: [PATCH] atol in test_call_gemm (#14480) flaky --- test/unit/test_call.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/unit/test_call.py b/test/unit/test_call.py index 473b2962bc..1c5ef3829a 100644 --- a/test/unit/test_call.py +++ b/test/unit/test_call.py @@ -42,7 +42,7 @@ class TestCall(unittest.TestCase): b = Tensor.randn(K, N) Tensor.realize(a, b) c = Tensor.call(a, b, fxn=a.as_param(0) @ b.as_param(1)) - np.testing.assert_allclose(c.numpy(), a.numpy() @ b.numpy(), rtol=1e-5) + np.testing.assert_allclose(c.numpy(), a.numpy() @ b.numpy(), rtol=1e-5, atol=1e-6) @unittest.skip("needs GEMM on mixins") def test_call_gemm_uop(self): @@ -56,7 +56,7 @@ class TestCall(unittest.TestCase): y = UOp.param(1, dtypes.float, shape=(K, N)) c = Tensor.call(a, b, fxn=x@y) - np.testing.assert_allclose(c.numpy(), a.numpy() @ b.numpy(), rtol=1e-5) + np.testing.assert_allclose(c.numpy(), a.numpy() @ b.numpy(), rtol=1e-5, atol=1e-6) if __name__ == '__main__': unittest.main()