github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-07 03:00:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

Revert "use movement ops [pr] (#8222)" (#8224)

Browse Source 
This reverts commit 0d26c970ba.
This commit is contained in:
George Hotz
2024-12-13 14:10:47 -08:00
committed by GitHub
parent 0d26c970ba
commit da19c37f0a
5 changed files with 13 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/imported/test_indexing.py
View File

@@ -1350,7 +1350,7 @@ class TestNumpy(unittest.TestCase):
# Empty tuple index creates a view
a = Tensor([1, 2, 3])
numpy_testing_assert_equal_helper(a[()], a)
#self.assertEqual(data_ptr(a[()]), data_ptr(a))
self.assertEqual(data_ptr(a[()]), data_ptr(a))
# TODO jax supports empty tensor indexing
@unittest.skip("empty tensor indexing not supported")
@@ -1372,7 +1372,7 @@ class TestNumpy(unittest.TestCase):
self.assertIsNot(a[...], a)
numpy_testing_assert_equal_helper(a[...], a)
# `a[...]` was `a` in numpy <1.9.
#numpy_testing_assert_equal_helper(data_ptr(a[...]), data_ptr(a))
numpy_testing_assert_equal_helper(data_ptr(a[...]), data_ptr(a))
# Slicing with ellipsis can skip an
# arbitrary number of dimensions