tinygrad/tinygrad
2
0
Fork 3
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-08-29 07:36:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

move in optimize

Browse Source
This commit is contained in:
geohot
2025-10-23 16:47:45 +08:00
parent ad6dd64fee
commit 6c66cbe5a1
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tinygrad/codegen/__init__.py
+3 -3
View File
@@ -19,13 +19,13 @@ from tinygrad.codegen.late.control_flow import CFGContext, pm_add_ends, pm_split
def full_rewrite_to_sink(sink:UOp, ren:Renderer|None=None, optimize:bool=True) -> UOp:
if ren is None: ren = Renderer()
# TODO: fix expander remove this
sink = graph_rewrite(sink, pm_add_buffers_local, name="add locals early")
# first we optimize
if optimize:
if QUANTIZE and ren.device in {"CPU", "DSP"}: sink = graph_rewrite(sink, pm_quant, name="quantize")
# TODO: fix expander and remove this
sink = graph_rewrite(sink, pm_add_buffers_local, name="add locals early")
# split ranges
sink = graph_rewrite(sink, pm_split_ranges+pm_flatten_range, ctx={}, name="split ranges")