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

fix getitem with leading None (#4943)

Browse Source 
i think all None handling can be unified and remove the calc_dim in advanced indexing
This commit is contained in:
chenyu
2024-06-13 11:23:40 -04:00
committed by GitHub
parent e63701fbd4
commit 74586bc339
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/test_ops.py
View File

@@ -1684,11 +1684,6 @@ class TestOps(unittest.TestCase):
helper_test_op([(2,5,6,5,3,4)], lambda x: x[None,b,c,d,None], lambda x: x[None,j,k,o,None])
helper_test_op([(2,5,6,5,3,4)], lambda x: x[a,:,None,d,e], lambda x: x[i,:,None,o,p])
helper_test_op([(2,5,6,5,3,4)], lambda x: x[None,None,None,None,None], lambda x: x[None,None,None,None,None])
# TODO: fix
@unittest.expectedFailure
def test_slice_fancy_indexing_dim_leading_nones(self):
a,b,c,d,e,i,j,k,o,p = self._get_index_randoms() # noqa
helper_test_op([(2,5,6,5,3,4)], lambda x: x[None,None,b,c,d,e], lambda x: x[None,None,j,k,o,p])
def test_slice_fancy_indexing_dim_inject_and_collapse(self):