mawei/tinygrad
1
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-13 00:15:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

assert __setitem__ if used other than disk (#3972)

Browse Source 
* assert `__setitem__` if used other than disk

* that is not implemented
This commit is contained in:
chenyu
2024-03-28 12:16:38 -04:00
committed by GitHub
parent 4b95350c41
commit bfcaa2f70e
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/imported/test_indexing.py
View File

@@ -1525,7 +1525,8 @@ class TestNumpy(unittest.TestCase):
def test_broaderrors_indexing(self):
a = Tensor.zeros(5, 5)
self.assertRaises(IndexError, a.__getitem__, ([0, 1], [0, 1, 2]))
self.assertRaises(IndexError, a.__setitem__, ([0, 1], [0, 1, 2]), 0)
# TODO setitem
# self.assertRaises(IndexError, a.__setitem__, ([0, 1], [0, 1, 2]), 0)
# TODO setitem
'''

4
tinygrad/tensor.py
View File

@@ -505,7 +505,9 @@ class Tensor:
ret = ret.permute(ret_dims[first_dim:first_dim+max_idx_dim] + ret_dims[:first_dim] + ret_dims[first_dim+max_idx_dim:])
return ret
def __setitem__(self,indices,v): return self.__getitem__(indices).assign(v)
def __setitem__(self,indices,v):
if isinstance(self.device, str) and self.device.startswith("DISK"): return self.__getitem__(indices).assign(v)
raise NotImplementedError("not implemented yet")
# NOTE: using slice is discouraged and things should migrate to pad and shrink
def slice(self, arg:Sequence[Optional[Tuple[int, sint]]], value:float=0) -> Tensor: