github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-09 15:08:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

multitensor start (#2676)

Browse Source 
* multitensor work

* early gen fixes the tests

* atol for flaky test
This commit is contained in:
George Hotz
2023-12-07 17:07:05 -08:00
committed by GitHub
parent 4b01839774
commit 4164d0ebbd
8 changed files with 118 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/external/external_test_opt.py vendored
View File

@@ -166,7 +166,7 @@ class TestOpt(unittest.TestCase):
with CLCache(allowed=1):
d = a * b + c
d.realize()
np.testing.assert_allclose(d.numpy(), na*nb+nc, rtol=1e-5)
np.testing.assert_allclose(d.numpy(), na*nb+nc, rtol=1e-5, atol=1e-7)
def test_fold_reduce_elementwise(self):
img = Tensor.ones(32)