github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-02-12 07:35:16 -05:00
Code Issues Packages Projects Releases Wiki Activity

const copy folding spec + multi.py behavior [pr] (#8436)

Browse Source 
* const copy folding spec + multi behavior [pr]

* copy from clang, move multi test
This commit is contained in:
qazal
2024-12-29 17:12:13 +02:00
committed by GitHub
parent a44cd1e6f7
commit 34987a03af
5 changed files with 32 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
test/test_schedule.py
View File

@@ -2163,5 +2163,19 @@ class TestConst(unittest.TestCase):
run_schedule(sched, var_vals)
self.assertEqual(a.tolist(), 3)
@unittest.skipIf(Device.DEFAULT == "CLANG", "tests copy from another device to clang")
class TestCopyFolding(unittest.TestCase):
def test_const_copy_is_free(self):
b = Tensor(1).to("CLANG")
check_schedule(b, 0, filter_sink=False)
assert b.item() == 1
def test_late_const_copy_folding(self):
a = Tensor.arange(3).realize()
zeros = Tensor.zeros(3).realize()
b = (a*zeros).to("CLANG")
run_schedule(check_schedule(b, 0, filter_sink=False))
self.assertListEqual(b.tolist(), [0, 0, 0])
if __name__ == '__main__':
unittest.main(verbosity=2)