commaai
/
tinygrad
mirror of https://github.com/commaai/tinygrad.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

enable symbolic_ops and jits test of two vars (#3513)

This commit is contained in:
chenyu 2024-02-27 11:17:46 -05:00 committed by GitHub
parent ea4b8e5b1f
commit 969b57f0fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_symbolic_jit.py
View File

@ -124,7 +124,6 @@ class TestSymbolicJit(unittest.TestCase):
np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6) np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6)
assert_jit_cache_len(jf, 1) assert_jit_cache_len(jf, 1)
@unittest.skip("two vars not supported")
def test_two_vars_plus1_ij(self): def test_two_vars_plus1_ij(self):
def f(a, b): return (a@b+1).realize() def f(a, b): return (a@b+1).realize()
jf = TinyJit(f) jf = TinyJit(f)
@ -139,7 +138,6 @@ class TestSymbolicJit(unittest.TestCase):
np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6) np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6)
assert_jit_cache_len(jf, 1) assert_jit_cache_len(jf, 1)
@unittest.skip("two vars not supported")
def test_two_vars_plus1_ji(self): def test_two_vars_plus1_ji(self):
def f(a, b): return (a@b+1).realize() def f(a, b): return (a@b+1).realize()
jf = TinyJit(f) jf = TinyJit(f)

2
test/test_symbolic_ops.py
View File

@ -108,7 +108,6 @@ class TestSymbolicOps(unittest.TestCase):
expected = f(a, b).numpy() expected = f(a, b).numpy()
np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6) np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6)
@unittest.skip("two vars not supported")
def test_two_vars_plus1_ij(self): def test_two_vars_plus1_ij(self):
def f(a, b): return (a@b+1).realize() def f(a, b): return (a@b+1).realize()
for i in range(1, 5): for i in range(1, 5):
@ -121,7 +120,6 @@ class TestSymbolicOps(unittest.TestCase):
expected = f(a, b).numpy() expected = f(a, b).numpy()
np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6) np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6)
@unittest.skip("two vars not supported")
def test_two_vars_plus1_ji(self): def test_two_vars_plus1_ji(self):
# reverse the order of variables # reverse the order of variables
def f(a, b): return (a@b+1).realize() def f(a, b): return (a@b+1).realize()