From 5820940d98920c59433e87bd29522c96142ad15a Mon Sep 17 00:00:00 2001 From: chenyu Date: Sat, 10 Aug 2024 16:10:03 -0400 Subject: [PATCH] more relax rtol for test_arange_fuse_grouped_children (#6027) one more https://github.com/chenyuxyz/tinygrad/actions/runs/10334072657/job/28607120462 --- test/test_schedule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_schedule.py b/test/test_schedule.py index 758b752881..3ab76209c6 100644 --- a/test/test_schedule.py +++ b/test/test_schedule.py @@ -1567,7 +1567,7 @@ class TestIndexing(unittest.TestCase): self.check_schedule([out0, out1], 1) r_ref = (X.numpy()+np.arange(16).reshape(4, 4)).sum() np.testing.assert_allclose(out0.numpy(), r_ref+2, rtol=2e-7) - np.testing.assert_allclose(out1.numpy(), r_ref+3) + np.testing.assert_allclose(out1.numpy(), r_ref+3, rtol=2e-7) @unittest.expectedFailure def test_fold_arange_view(self):