nimlgen/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Activity

fix

Browse Source
This commit is contained in:
ttomsa
2025-10-09 22:08:32 +01:00
parent 8b63413e7b
commit 7ea44c60f6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/codegen/late/control_flow.py
+2 -2
View File
@@ -47,9 +47,9 @@ def schedule(lst:list[UOp]) -> list[UOp]:
def add_endrange(x:UOp):
if not ((x.op is Ops.LOAD and x.src[-1].op is Ops.STORE) or all(s.op is Ops.STORE and any(n.op is Ops.RANGE for n in s.src) for s in x.src)):
return None
src: list[tuple[UOp, ...]] = []
src: list[UOp] = []
for k,g in groupby(x.src, key=lambda k: tuple(dedup(s for s in k.src if s.op is Ops.RANGE))):
if not k: src.extend(tuple(g))
if not k: src.extend(g)
else: src.extend(reduce(lambda acc,rng: (UOp(Ops.ENDRANGE, src=(rng,) + acc),), reversed(k), tuple(g)))
return x.replace(src=tuple(src))