github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

new test for e2e compile failures (#6101)

Browse Source 
* new test for e2e compile failures

* fix bug
This commit is contained in:
George Hotz
2024-08-15 18:56:22 -07:00
committed by GitHub
parent e4a7869893
commit c850e03758
2 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
test/test_compile_failures.py Normal file
View File

@@ -0,0 +1,13 @@
import unittest
from tinygrad import Tensor
from tinygrad.engine.realize import lower_schedule
class TestCompileFailures(unittest.TestCase):
def compile(self, out:Tensor):
for _ in lower_schedule(out.schedule()): pass
def test_interpolate_atari(self):
self.compile(Tensor.empty(210, 160, dtype='uint8').interpolate((64, 64)))
if __name__ == '__main__':
unittest.main()