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

update docs, remove corealize (#4264)

Browse Source 
* update docs, remove corealize

* handle 0 line count

* tensor schedule
This commit is contained in:
George Hotz
2024-04-23 12:05:29 +04:00
committed by GitHub
parent 9b7efa72ea
commit 967638f0d5
11 changed files with 61 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_assign.py
View File

@@ -171,7 +171,7 @@ class TestAssign(unittest.TestCase):
b = Tensor.full((4,), 3).contiguous().realize()
a += b
b += a
Tensor.corealize([a,b])
Tensor.realize(a,b)
np.testing.assert_allclose(a.numpy(), 5)
np.testing.assert_allclose(b.numpy(), 8)
@@ -183,7 +183,7 @@ class TestAssign(unittest.TestCase):
c = a+9
a += b
b += c
Tensor.corealize([a,b])
Tensor.realize(a,b)
np.testing.assert_allclose(a.numpy(), 2+3)
np.testing.assert_allclose(b.numpy(), 3+2+9)