diff --git a/noxfile.py b/noxfile.py index 31cb10dc5..cee275ef1 100644 --- a/noxfile.py +++ b/noxfile.py @@ -42,7 +42,7 @@ "packaging", "protobuf", ) -ONNX_IR = "onnx_ir==0.1.1" +ONNX_IR = "onnx_ir==0.1.3" ONNX_IR_MAIN = "git+https://github.com/onnx/ir-py.git@main#egg=onnx_ir" diff --git a/onnxscript/optimizer/_optimizer.py b/onnxscript/optimizer/_optimizer.py index e017ee205..ba03a44d9 100644 --- a/onnxscript/optimizer/_optimizer.py +++ b/onnxscript/optimizer/_optimizer.py @@ -56,6 +56,7 @@ def optimize_ir( common_passes.CommonSubexpressionEliminationPass(), common_passes.LiftConstantsToInitializersPass(), common_passes.LiftSubgraphInitializersToMainGraphPass(), + common_passes.DeduplicateInitializersPass(), ] if inline: # Inline all functions first before optimizing