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

Move var_vals from ShapeTracker to LazyBuffer (#1819)

Browse Source
This commit is contained in:
chenyu
2023-09-08 09:25:10 -07:00
committed by GitHub
parent 7ac65a93b4
commit ebcda8a714
11 changed files with 63 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_symbolic_jit.py
View File

@@ -171,7 +171,7 @@ class TestSymbolicJit(unittest.TestCase):
for i in range(1, 5):
a = Tensor.rand(7, 11)
symbolic = a.shrink(((3,5),(vi,vi+2)))
symbolic.lazydata.st.var_vals[vi] = i
symbolic.lazydata.var_vals[vi] = i
symbolic = jf(symbolic).numpy()
expected = f(a.shrink(((3,5),(i,i+2)))).numpy()
np.testing.assert_allclose(symbolic, expected, atol=1e-6, rtol=1e-6)