github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

ttomsa linearizer

Browse Source
This commit is contained in:
George Hotz
2025-10-10 15:08:28 +08:00
parent b38f3595bf
commit 9ec77ff587
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tinygrad/uop/ops.py
View File

@@ -233,7 +233,10 @@ class UOp(MathTrait, metaclass=UOpMetaClass):
@recursive_property
def _ranges(self) -> dict[UOp, None]:
ret: dict[UOp, None] = {}
if self.op in range_start.keys():
if self.op is Ops.ENDRANGE:
for s in self.src[1:]: ret.update(s.ranges)
assert self.src[0] not in ret
elif self.op in range_start.keys():
for s in self.src[:range_start[self.op]]: ret.update(s.ranges)
for s in UOp.sink(*self.src[range_start[self.op]:]).ranges:
if s in ret: del ret[s]