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

check shrink is actually smaller

Browse Source
This commit is contained in:
George Hotz
2023-03-09 12:59:45 -08:00
parent dbbaa0bdd7
commit de1b6d3e08
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/shape/__init__.py
View File

@@ -182,7 +182,7 @@ class ShapeTracker:
def shrink(self, arg : Tuple[Tuple[int, int], ...]) -> ShapeTracker:
assert isinstance(arg, tuple)
assert len(arg) == len(self.shape)
assert all((b>=0 and e<=s) for s,(b,e) in zip(self.shape,arg)) and len(arg) == len(self.shape)
self._resize(arg)
return self