Skip to content

Commit

Permalink
allbufs are base (tinygrad#4605)
Browse files Browse the repository at this point in the history
  • Loading branch information
Qazalin authored May 15, 2024
1 parent 067ff71 commit 36d2ac6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tinygrad/engine/schedule.py
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,7 @@ def _graph_schedule(outs:List[LazyBuffer], seen:Set[LazyBuffer]) -> Tuple[Defaul
# find all reduces, and pair them to a elementwise op. if they can't be cleanly paired, force realize the reduce (or a contig child)
reduce_for_op: Dict[LazyBuffer, LazyBuffer] = {}
for r in allbufs:
if r != r.base or r.op not in ReduceOps or r in realizes: continue
if r.op not in ReduceOps or r in realizes: continue

group: Set[LazyBuffer] = set()
_recursive_group(r, r.st, r, children, realizes, reduce_for_op, group)
Expand Down

0 comments on commit 36d2ac6

Please sign in to comment.