tenstorrent
diff --git a/‎torch_ttnn/backend.py
Lines changed: 1 addition & 0 deletions b/‎torch_ttnn/backend.py
Lines changed: 1 addition & 0 deletions
diff --git a/‎torch_ttnn/passes/mm_fusion_pass.py
Lines changed: 0 additions & 58 deletions b/‎torch_ttnn/passes/mm_fusion_pass.py
Lines changed: 0 additions & 58 deletions
@@ -148,6 +148,7 @@ def aten_backend(
             passes_with_graphviz.append(
                 GraphvizPass(f"metrics/{option.metrics_path}/{idx + 1:02d}.{passes[idx].__class__.__name__}")
             )
+        passes = passes_with_graphviz
 
     pm = PassManager(passes=passes)
     gm, modified = pm(gm)
Original file line number	Diff line number	Diff line change
`@@ -148,6 +148,7 @@ def aten_backend(`
`148`	`148`	`passes_with_graphviz.append(`
`149`	`149`	`GraphvizPass(f"metrics/{option.metrics_path}/{idx + 1:02d}.{passes[idx].__class__.__name__}")`
`150`	`150`	`)`
	`151`	`+ passes = passes_with_graphviz`
`151`	`152`
`152`	`153`	`pm = PassManager(passes=passes)`
`153`	`154`	`gm, modified = pm(gm)`