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

remove unused unstride

Browse Source
This commit is contained in:
George Hotz
2022-06-15 20:03:43 -07:00
parent ff648e9510
commit 3667200df5
1 changed files with 0 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tinygrad/shapetracker.py
View File

@@ -100,14 +100,6 @@ class ShapeTracker:
offset = sum([(s-1)*z for s,z,m in zip(self.shape,old_strides,mul) if m < 0])
self.views.append(View(new_shape, strides, offset))
# TODO: is there a better name for this?
# TODO: this shouldn't be in shapetracker, this should be in mlops
def unstride(self, *mul):
start_shape = self.shape
self.reshape(*reduce(lambda x,y: x+[y,1], start_shape, []))
self.slice([(0,x) for x in chain(*zip(start_shape, mul))])
self.reshape([s*m for s,m in zip(start_shape, mul)])
# TODO: this is a special case of slice with strides, remove it
def flip(self, *axis):
self.stride(*[-1 if i in axis else 1 for i in range(len((self.shape)))])