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

Revert "limit metal buffers and revert the 207 fix (try 2) (#2088)"

Browse Source 
This reverts commit 5e24dc5a95.
This commit is contained in:
George Hotz
2023-10-16 17:50:11 -07:00
parent d0aaf7d83b
commit e4846771b2
2 changed files with 4 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/test_schedule.py
View File

@@ -326,17 +326,5 @@ class TestSchedule(unittest.TestCase):
out = x.to('cpu')
check_schedule(out, 0, filter_loadops=False)
@unittest.skipUnless(Device.DEFAULT == "METAL", "only for metal")
def test_metal_limit_buffers(self):
t = sum([Tensor([1,2,3,4]) for _ in range(40)])
for si in t.lazydata.schedule():
assert len(si.inputs) <= 30
@unittest.skipUnless(Device.DEFAULT == "METAL", "only for metal")
def test_metal_dont_limit_same_buffers(self):
bt = Tensor(list(range(1, 100)))
out = sum([bt[i:i+2] for i in range(1,40)])
check_schedule(out, 1)
if __name__ == '__main__':
unittest.main(verbosity=2)