[DORT] Update import path (microsoft#14605)

Follow up changes from https://github.com/pytorch/pytorch/pull/93409/files for fixing DORT CI failures.
intel · Feb 15, 2023 · 9d2144c · 9d2144c
1 parent 287304f
commit 9d2144c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/orttraining/orttraining/python/training/torchdynamo/register_backend.py b/orttraining/orttraining/python/training/torchdynamo/register_backend.py
@@ -4,7 +4,7 @@
 # --------------------------------------------------------------------------
 
 from functorch.compile import min_cut_rematerialization_partition
-from torch._dynamo.optimizations.training import aot_autograd
+from torch._dynamo.backends.common import aot_autograd
 from .ort_backend import OrtBackend