diff --git a/tinygrad/codegen/uopgraph.py b/tinygrad/codegen/uopgraph.py index f9881b4073..d16207d93f 100644 --- a/tinygrad/codegen/uopgraph.py +++ b/tinygrad/codegen/uopgraph.py @@ -587,7 +587,7 @@ class UOpGraph: while queue: p,x = heapq.heappop(queue) if DEBUG >= 7: print(p,x) - if x.op is UOps.DEFINE_ACC and len(x.src) > 1: + if x.op is UOps.DEFINE_ACC: idx = min([self._uops.index(l) for l in x.src if l.op is UOps.RANGE]) self._uops.insert(idx, x) else: self._uops.append(x)