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)