From 33d44f00aea2a6cb1edb624c4367b94abca3a136 Mon Sep 17 00:00:00 2001 From: George Hotz <72895+geohot@users.noreply.github.com> Date: Wed, 24 Jul 2024 09:43:09 -0700 Subject: [PATCH] first fold, then expand (#5673) Co-authored-by: chenyu --- tinygrad/codegen/uopgraph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tinygrad/codegen/uopgraph.py b/tinygrad/codegen/uopgraph.py index f0d718921c..0ec5e6acaf 100644 --- a/tinygrad/codegen/uopgraph.py +++ b/tinygrad/codegen/uopgraph.py @@ -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)