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

test pushing through different expands in 1 kernel (#5963)

Browse Source 
* test pushing through different expands in 1 kernel

* realize eye

* back to test_example_matmul
This commit is contained in:
qazal
2024-08-08 00:33:18 +08:00
committed by GitHub
parent 564a352194
commit 7677361d90
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
test/test_schedule.py
View File

@@ -1496,5 +1496,14 @@ class TestIndexing(unittest.TestCase):
self.check_schedule([r], 1)
np.testing.assert_allclose(r.numpy(), (X.numpy()+np.arange(16).reshape(4, 4)).sum(1, keepdims=True))
@unittest.expectedFailure
def test_multiview_arange_children(self):
X = Tensor.randn(2,3,4,4).numpy()
with Context(FUSE_ARANGE=1):
compare = Tensor(X).interpolate(size=(2, 2), mode="linear").numpy()
with Context(FUSE_ARANGE=0, GRAPH=0, SAVE_SCHEDULE=1):
ref = Tensor(X).interpolate(size=(2, 2), mode="linear").numpy()
np.testing.assert_allclose(ref, compare, atol=1e-5, rtol=1e-6)
if __name__ == '__main__':
unittest.main(verbosity=2)