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

late loads

Browse Source
This commit is contained in:
geohot
2026-07-07 15:35:58 -07:00
parent 5a4648924b
commit ee70cffcee
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/codegen/__init__.py
+2 -2
View File
@@ -304,13 +304,13 @@ def full_rewrite_to_sink(ast:UOp, ren:Renderer, optimize:bool=True) -> UOp:
# **** optimizations are done, now we lower to actual code ****
sink = graph_rewrite(sink, symbolic_simple+unbroadcast+pm_add_loads, name="*** unbroadcast / add loads")
sink = graph_rewrite(sink, mop_cleanup+symbolic_simple+unbroadcast, name="*** unbroadcast")
# devectorize
sink = graph_rewrite(sink, symbolic_simple+devectorizer2, ctx=ren, name="devectorize2")
# simplify indexing
sink = graph_rewrite(sink, indexing_simplify, name="simplify load/store indexing")
sink = graph_rewrite(sink, pm_add_loads+indexing_simplify, name="simplify load/store indexing")
# some coalesing misses without this
sink = graph_rewrite(sink, sym, name="early symbolic")