[DORT] Update import path (#14605)

Follow up changes from
https://github.com/pytorch/pytorch/pull/93409/files for fixing DORT CI
failures.
This commit is contained in:
Wei-Sheng Chin 2023-02-08 19:54:06 -08:00 committed by GitHub
parent 96b95a24ee
commit 875a7791bf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -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