[ONNX] Update onnx to 1.18 (pytorch#153746)

Update onnx python package to 1.18.

Pull Request resolved: pytorch#153746
Approved by: https://github.com/titaiwangms, https://github.com/cyyever, https://github.com/malfet

Author: justinchuby

.ci/docker/common/install_onnx.sh

@@ -31,8 +31,7 @@ pip_install \
 pip_install coloredlogs packaging
 
 pip_install onnxruntime==1.18.1
-pip_install onnx==1.17.0
-pip_install onnxscript==0.2.2 --no-deps
+pip_install onnxscript==0.2.6 --no-deps
 # required by onnxscript
 pip_install ml_dtypes
 

.ci/docker/requirements-ci.txt

@@ -169,7 +169,7 @@ pillow==11.0.0
 protobuf==5.29.4
 #Description:  Google's data interchange format
 #Pinned versions: 5.29.4
-#test that import: test_tensorboard.py
+#test that import: test_tensorboard.py, test/onnx/*
 
 psutil
 #Description: information on running processes and system utilization
@@ -337,12 +337,12 @@ sympy==1.13.3
 #Pinned versions:
 #test that import:
 
-onnx==1.17.0
-#Description: Required by mypy and test_public_bindings.py when checking torch.onnx._internal
+onnx==1.18.0
+#Description: Required by onnx tests, and mypy and test_public_bindings.py when checking torch.onnx._internal
 #Pinned versions:
 #test that import:
 
-onnxscript==0.2.2
+onnxscript==0.2.6
 #Description: Required by mypy and test_public_bindings.py when checking torch.onnx._internal
 #Pinned versions:
 #test that import:

test/onnx/exporter/test_small_models_e2e.py

@@ -224,7 +224,7 @@ class Float4Module(torch.nn.Module):
             def forward(self):
                 return torch.empty([1], dtype=torch.float4_e2m1fn_x2)
 
-        onnx_program = self.export(Float4Module())
+        onnx_program = self.export(Float4Module(), optimize=False)
         output = onnx_program.model.graph.outputs[0]
         self.assertEqual(output.dtype, ir.DataType.FLOAT4E2M1)
         # The shape is [*shape, 2] because ONNX stores the shape of the unpacked tensor

torch/onnx/_internal/exporter/_onnx_program.py

@@ -81,7 +81,7 @@ def _set_graph_outputs(
         graph: The graph to set the outputs for.
         outputs: The outputs to set.
     """
-    original_outputs = graph.outputs.copy()
+    original_outputs = list(graph.outputs)
     graph.outputs.clear()
     graph.outputs.extend(outputs)
     try:
@@ -102,7 +102,7 @@ def _create_value_mapping(graph: ir.Graph) -> dict[str, ir.Value]:
     Returns:
         A dictionary mapping names to values.
     """
-    values = {}
+    values: dict[str, ir.Value] = {}
     values.update(graph.initializers)
     # The names of the values can be None or "", which we need to exclude
     for input in graph.inputs:

torch/onnx/_internal/exporter/_verification.py

@@ -224,7 +224,7 @@ def _create_value_mapping(graph: ir.Graph) -> dict[str, ir.Value]:
     Returns:
         A dictionary mapping names to values.
     """
-    values = {}
+    values: dict[str, ir.Value] = {}
     values.update(graph.initializers)
     # The names of the values can be None or "", which we need to exclude
     for input in graph.inputs:
@@ -322,6 +322,7 @@ def run_node(self, n: torch.fx.Node) -> Any:
                 "Failed to compute value for node %s: %s",
                 node_name,
                 e,
+                exc_info=True,
             )
             return result
         info = VerificationInfo.from_tensors(