mirror of https://github.com/commaai/tinygrad.git
enable symbolic_ops and jits test of two vars (#3513)
This commit is contained in:
parent
ea4b8e5b1f
commit
969b57f0fe
|
@ -124,7 +124,6 @@ class TestSymbolicJit(unittest.TestCase):
|
|||
np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6)
|
||||
assert_jit_cache_len(jf, 1)
|
||||
|
||||
@unittest.skip("two vars not supported")
|
||||
def test_two_vars_plus1_ij(self):
|
||||
def f(a, b): return (a@b+1).realize()
|
||||
jf = TinyJit(f)
|
||||
|
@ -139,7 +138,6 @@ class TestSymbolicJit(unittest.TestCase):
|
|||
np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6)
|
||||
assert_jit_cache_len(jf, 1)
|
||||
|
||||
@unittest.skip("two vars not supported")
|
||||
def test_two_vars_plus1_ji(self):
|
||||
def f(a, b): return (a@b+1).realize()
|
||||
jf = TinyJit(f)
|
||||
|
|
|
@ -108,7 +108,6 @@ class TestSymbolicOps(unittest.TestCase):
|
|||
expected = f(a, b).numpy()
|
||||
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 f(a, b): return (a@b+1).realize()
|
||||
for i in range(1, 5):
|
||||
|
@ -121,7 +120,6 @@ class TestSymbolicOps(unittest.TestCase):
|
|||
expected = f(a, b).numpy()
|
||||
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):
|
||||
# reverse the order of variables
|
||||
def f(a, b): return (a@b+1).realize()
|
||||
|
|
Loading…
Reference in New Issue