Add OutputFixPass and NameFixPass in optimize (#2779)

justinchuby · web-flow · commit 55ed939da8f7 · 2026-01-26T09:32:38.000-08:00
Fix pytorch/pytorch#168211 Fix pytorch/pytorch#167848 Partially address pytorch/pytorch#173076 Signed-off-by: Justin Chu <justinchuby@users.noreply.github.com>
diff --git a/onnxscript/optimizer/_optimizer.py b/onnxscript/optimizer/_optimizer.py
@@ -64,6 +64,8 @@ def optimize_ir(
         common_passes.LiftSubgraphInitializersToMainGraphPass(),
         common_passes.DeduplicateInitializersPass(),
         common_passes.CommonSubexpressionEliminationPass(),
+        common_passes.OutputFixPass(),
+        common_passes.NameFixPass(),
     ]
     if inline:
         # Inline all functions first before optimizing

Original file line number	Diff line number	Diff line change
`@@ -64,6 +64,8 @@ def optimize_ir(`
`64`	`64`	`common_passes.LiftSubgraphInitializersToMainGraphPass(),`
`65`	`65`	`common_passes.DeduplicateInitializersPass(),`
`66`	`66`	`common_passes.CommonSubexpressionEliminationPass(),`
	`67`	`+ common_passes.OutputFixPass(),`
	`68`	`+ common_passes.NameFixPass(),`
`67`	`69`	`]`
`68`	`70`	`if inline:`
`69`	`71`	`# Inline all functions first before optimizing`