From 9914129a07315311b522208db94b54577dd7e739 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 5 Jul 2023 11:08:38 +0800 Subject: [PATCH] fix(imperative): fix the circular import in trace GitOrigin-RevId: 4c0ed6fc55e9af9fd245997c6277363e2b6c2cab --- imperative/python/megengine/jit/xla_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/imperative/python/megengine/jit/xla_backend.py b/imperative/python/megengine/jit/xla_backend.py index 6e9e48e70..992caedcb 100644 --- a/imperative/python/megengine/jit/xla_backend.py +++ b/imperative/python/megengine/jit/xla_backend.py @@ -4,7 +4,6 @@ from ..core._imperative_rt import CompNode from ..core._imperative_rt.core2 import Tensor as RawTensor from ..core._trace_option import set_use_xla_backend from ..device import get_default_device -from ..distributed import get_mm_server_addr, is_distributed from ..utils.dlpack import from_dlpack, to_dlpack from .tracing import trace @@ -65,6 +64,7 @@ class xla_trace(trace): from ..utils.module_utils import get_expand_structure from ..xla.device import get_xla_backend_and_device from ..tensor import Tensor + from ..distributed import get_mm_server_addr, is_distributed assert self.traced if self.overall: -- GitLab