github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-26 15:28:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

first fold, then expand (#5673)

Browse Source 
Co-authored-by: chenyu <chenyu@fastmail.com>
This commit is contained in:
George Hotz
2024-07-24 09:43:09 -07:00
committed by GitHub
parent b7b4c7844f
commit 33d44f00ae
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tinygrad/codegen/uopgraph.py
View File

@@ -501,7 +501,7 @@ class UOpGraph:
# expand
UOpGraph.cnt += 1
if UOpGraph.cnt != getenv("DEBUG_EXPAND", 0): sink = graph_rewrite(sink, expander+self.folder)
if UOpGraph.cnt != getenv("DEBUG_EXPAND", 0): sink = graph_rewrite(sink, self.folder+expander)
# for PTX only
if extra_pm: sink = graph_rewrite(sink, self.folder+extra_pm)