From 140c044efd3ac1b6c40afda6aeef266de8dacfdb Mon Sep 17 00:00:00 2001 From: zpcore Date: Thu, 18 Jul 2024 01:00:34 +0000 Subject: [PATCH] update module --- torch_xla/core/xla_model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/torch_xla/core/xla_model.py b/torch_xla/core/xla_model.py index f973b180fa0..6bf2248dfe9 100644 --- a/torch_xla/core/xla_model.py +++ b/torch_xla/core/xla_model.py @@ -20,7 +20,7 @@ import torch_xla.utils.closures as xc import os from torch_xla.experimental.deprecation import deprecated -import torch_xla._internal.utils as iutils +import torch_xla._internal.utils as _utils _DEVICES = xu.LazyProperty(lambda: torch_xla._XLAC._xla_get_devices()) @@ -42,7 +42,8 @@ XLA_LIB = Library("xla", "DEF") -parse_xla_device = deprecated(torch_xla.core, iutils.parse_xla_device) +from . import xla_model as this_module +parse_xla_device = deprecated(this_module, _utils.parse_xla_device) def _init_world_size_ordinal():