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

ugh, don't use os

Browse Source
This commit is contained in:
George Hotz
2023-01-25 19:41:21 -08:00
parent b1dec64815
commit 60acb2641f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tinygrad/shape/__init__.py
View File

@@ -64,9 +64,9 @@ class ZeroView:
def expr_node(self, valid, idx):
expr, acc = [valid] if valid is not None else [], 1
for os,ns,(x,y) in list(zip(self.old_shape, self.shape, self.arg))[::-1]:
for s,ns,(x,y) in list(zip(self.old_shape, self.shape, self.arg))[::-1]:
base = ((idx//acc) % ns) + x
expr += ([base >= 0] if x < 0 else []) + ([base < os] if y > os else [])
expr += ([base >= 0] if x < 0 else []) + ([base < s] if y > s else [])
acc *= ns
return Variable.ands(expr)