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

sort vars in jit when building expected input args (#4990)

Browse Source 
* sort vars in jit when building expected input args

fixed symbolic jit bugs with two variables.

* sort in clanggraph

* space

* one more
This commit is contained in:
chenyu
2024-06-16 15:55:51 -04:00
committed by GitHub
parent 71aad183fd
commit 72c9b22833
5 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_symbolic_jit.py
View File

@@ -212,7 +212,6 @@ class TestSymbolicJit(unittest.TestCase):
expected = a.mean(1).numpy()
np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6)
@unittest.skip("failed for some")
def test_mean_2d(self):
def f(a): return a.mean().realize()
def f0(a): return a.mean(0).realize()
@@ -265,7 +264,6 @@ class TestSymbolicJit(unittest.TestCase):
expected = a.var(1).numpy()
np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6)
@unittest.skip("failed for some")
def test_var_2d(self):
def f(a): return a.var().realize()
def f0(a): return a.var(0).realize()