Fix DORT CI according to torch-nightly changes (#2125)

Author: justinchuby

onnxscript/tools/training_helper.py

@@ -3,16 +3,13 @@
 # Licensed under the MIT License.
 # --------------------------------------------------------------------------
 import torch
-from torch.onnx import ExportOptions
-from torch.onnx import _OrtBackend as OrtBackend
-from torch.onnx import _OrtBackendOptions as OrtBackendOptions
+from torch.onnx import _OrtBackend, _OrtBackendOptions
 
 
-def make_aot_ort(dynamic: bool = False):
+def make_aot_ort():
     """Implements an autograd backend for torch.compile based on onnxrt backend."""
-    export_options = ExportOptions(dynamic_shapes=dynamic)
-    options = OrtBackendOptions(export_options=export_options)
-    ort_backend = OrtBackend(options=options)
+    options = _OrtBackendOptions()
+    ort_backend = _OrtBackend(options=options)
     return ort_backend
 
 

onnxscript/tools/transformers_models/phi_test.py

@@ -89,7 +89,7 @@ def test_phi_dort_static(self):
         input_tensors = input_tensors_many[0]
         expected = model(*input_tensors)
 
-        local_aot_ort = onnxscript.tools.training_helper.make_aot_ort(dynamic=False)
+        local_aot_ort = onnxscript.tools.training_helper.make_aot_ort()
 
         compiled_model = torch.compile(
             copy.deepcopy(model),