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

failed test case when init jit with empty inputs (#13641)

Browse Source 
not related to bert grad acc, but still seems to be a bug
This commit is contained in:
chenyu
2025-12-10 22:03:06 -05:00
committed by GitHub
parent 51f3c9f615
commit 03600aef1e
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test/test_jit.py
View File

@@ -501,6 +501,18 @@ class TestJit(unittest.TestCase):
b = f(Tensor([2.0])) b = f(Tensor([2.0]))
assert abs((a - b).item()) > 0.5 assert abs((a - b).item()) > 0.5
def test_jit_init_with_empty_different_size(self):
@TinyJit
def f(x:Tensor) -> Tensor: return (x + 1).realize()
f(Tensor.empty(1))
f(Tensor.empty(1))
# TODO: this should fail since input has a different size
f(Tensor(2.0)).item()
# TODO: this should not fail, and should return 3
with self.assertRaises(AssertionError):
f(Tensor([2.0])).item()
@unittest.skip("Pending multioutput implementation #3607") @unittest.skip("Pending multioutput implementation #3607")
class TestMultioutputJit(unittest.TestCase): class TestMultioutputJit(unittest.TestCase):
def _test(self, f): def _test(self, f):