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

numpy device + pickle it (#4120)

Browse Source
This commit is contained in:
George Hotz
2024-04-09 13:19:30 -07:00
committed by GitHub
parent 1ef9c50fd7
commit ae849d12d7
7 changed files with 22 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_schedule.py
View File

@@ -350,7 +350,7 @@ class TestSchedule(unittest.TestCase):
def test_double_from(self):
x = Tensor([1,2,3,4])
out = x.to('ext')
out = x.to('npy')
check_schedule(out, 0, filter_loadops=False)
def test_pow_const_tensor_simplified(self):