relax rtol for test_arange_fuse_grouped_children (#6026)

flaky https://github.com/tinygrad/tinygrad/actions/runs/10333939631/job/28606831006?pr=6023
This commit is contained in:
chenyu
2024-08-10 15:49:11 -04:00
committed by GitHub
parent cf7d3c1eb8
commit 10374a2741

View File

@@ -1566,7 +1566,7 @@ class TestIndexing(unittest.TestCase):
out1 = r+3
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)
np.testing.assert_allclose(out0.numpy(), r_ref+2, rtol=2e-7)
np.testing.assert_allclose(out1.numpy(), r_ref+3)
@unittest.expectedFailure