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

preschedule all (#3875)

Browse Source
This commit is contained in:
George Hotz
2024-03-22 11:20:06 -07:00
committed by GitHub
parent d8c3f1894a
commit 86ee36e697
2 changed files with 18 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/test_assign.py
View File

@@ -100,6 +100,14 @@ class TestAssign(unittest.TestCase):
new = a + times_a
np.testing.assert_allclose(new.numpy(), 5)
@unittest.expectedFailure
def test_assign_diamond_possible(self):
a = Tensor.ones(4).contiguous().realize()
times_a = a*3
a.assign(Tensor.full((4,), 2.).contiguous())
new = a + (times_a+1).contiguous()
np.testing.assert_allclose(new.numpy(), 6)
def test_assign_diamond_alt(self):
a = Tensor.ones(4).contiguous().realize()
a.assign(Tensor.full((4,), 2.).contiguous())