From 875a7791bf610f9d5cc670ad20dfb58627406aba Mon Sep 17 00:00:00 2001 From: Wei-Sheng Chin Date: Wed, 8 Feb 2023 19:54:06 -0800 Subject: [PATCH] [DORT] Update import path (#14605) Follow up changes from https://github.com/pytorch/pytorch/pull/93409/files for fixing DORT CI failures. --- .../orttraining/python/training/torchdynamo/register_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/orttraining/orttraining/python/training/torchdynamo/register_backend.py b/orttraining/orttraining/python/training/torchdynamo/register_backend.py index a9450e119c..2830a10b4f 100644 --- 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