From ede701ef3bff828701c5f034d49ca1d02df1ac2f Mon Sep 17 00:00:00 2001 From: Ti-Tai Wang Date: Tue, 24 Jun 2025 22:03:37 +0000 Subject: [PATCH 1/2] enable DeduplicateInitializersPass --- onnxscript/optimizer/_optimizer.py | 1 + 1 file changed, 1 insertion(+) diff --git a/onnxscript/optimizer/_optimizer.py b/onnxscript/optimizer/_optimizer.py index 6044f35424..f6fbab5572 100644 --- a/onnxscript/optimizer/_optimizer.py +++ b/onnxscript/optimizer/_optimizer.py @@ -54,6 +54,7 @@ def optimize_ir( onnxscript.ir.passes.common.CommonSubexpressionEliminationPass(), onnxscript.ir.passes.common.LiftConstantsToInitializersPass(), onnxscript.ir.passes.common.LiftSubgraphInitializersToMainGraphPass(), + onnxscript.ir.passes.common.DeduplicateInitializersPass(), ] if inline: # Inline all functions first before optimizing From d4dd177038928626c490a2c5d47ca8254b0a2b3a Mon Sep 17 00:00:00 2001 From: Ti-Tai Wang Date: Wed, 25 Jun 2025 22:07:43 +0000 Subject: [PATCH 2/2] use onnx_ir --- onnxscript/optimizer/_optimizer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/onnxscript/optimizer/_optimizer.py b/onnxscript/optimizer/_optimizer.py index f6fbab5572..6fc68f6d16 100644 --- a/onnxscript/optimizer/_optimizer.py +++ b/onnxscript/optimizer/_optimizer.py @@ -4,6 +4,8 @@ import logging +import onnx_ir + import onnxscript.ir.passes.common from onnxscript import ir, rewriter from onnxscript.optimizer import _constant_folding @@ -54,7 +56,7 @@ def optimize_ir( onnxscript.ir.passes.common.CommonSubexpressionEliminationPass(), onnxscript.ir.passes.common.LiftConstantsToInitializersPass(), onnxscript.ir.passes.common.LiftSubgraphInitializersToMainGraphPass(), - onnxscript.ir.passes.common.DeduplicateInitializersPass(), + onnx_ir.passes.common.DeduplicateInitializersPass(), ] if inline: # Inline all functions first before optimizing