geohot/tinygrad
1
0
Fork 0
forked from tinygrad/tinygrad
Code Pull Requests Actions Activity

29 failures

Browse Source
This commit is contained in:
geohot
2026-02-19 12:36:09 +08:00
parent db2f41c92d
commit 619362530e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/engine/schedule.py
+1 -1
View File
@@ -147,7 +147,7 @@ def complete_create_schedule_with_vars(big_sink:UOp) -> tuple[dict[UOp, UOp], li
# new preschedule stuff
# we assign for any contiguous or anything on the sink that's not a BUFFER or a CONST
buffer_map: dict[UOp, UOp|None] = {x.base:None for x in big_sink.src if x.base.op not in {Ops.CONST, Ops.BUFFER}}
buffer_map: dict[UOp, UOp|None] = {x.base:None for x in big_sink.src if x.base.op not in {Ops.CONST, Ops.BUFFER, Ops.BIND, Ops.DEFINE_VAR}}
big_sink = graph_rewrite(big_sink, pm_build_buffer_map, ctx=buffer_map, bottom_up=True, name="contig to assign")
big_sink = graph_rewrite(big_sink, _remove_all_tags)
assert all(x is not None for x in buffer_map.values())