fix import

Signed-off-by: Masahiro Tanaka <[email protected]>

Author: tohtana

deepspeed/compile/backend.py

@@ -9,12 +9,12 @@
 import torch
 from torch.fx import Graph, GraphModule
 import torch.utils._pytree as pytree
-import torch._dynamo
-from torch._functorch.aot_autograd import aot_module_simplified
 
 try:
+    import torch._dynamo
     import torch._inductor.scheduler
     from functorch.compile import make_boxed_func
+    from torch._functorch.aot_autograd import aot_module_simplified
 except ImportError:
     pass