Lightning-AI
diff --git a/‎src/lightning_lite/utilities/__init__.py‎ b/‎src/lightning_lite/utilities/__init__.py‎
diff --git a/‎src/pytorch_lightning/utilities/xla_device.py‎ renamed to ‎src/lightning_lite/utilities/xla_device.py‎ b/‎src/pytorch_lightning/utilities/xla_device.py‎ renamed to ‎src/lightning_lite/utilities/xla_device.py‎
diff --git a/‎src/pytorch_lightning/utilities/imports.py‎
Lines changed: 1 addition & 1 deletion b/‎src/pytorch_lightning/utilities/imports.py‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎tests/tests_lite/utilities/__init__.py‎ b/‎tests/tests_lite/utilities/__init__.py‎
diff --git a/‎tests/tests_pytorch/utilities/test_xla_device_utils.py‎ renamed to ‎tests/tests_lite/utilities/test_xla_device_utils.py‎
Lines changed: 1 addition & 1 deletion b/‎tests/tests_pytorch/utilities/test_xla_device_utils.py‎ renamed to ‎tests/tests_lite/utilities/test_xla_device_utils.py‎
Lines changed: 1 addition & 1 deletion
@@ -147,7 +147,7 @@ def __repr__(self) -> str:
 _XLA_AVAILABLE: bool = _package_available("torch_xla")
 
 
-from pytorch_lightning.utilities.xla_device import XLADeviceUtils  # noqa: E402
+from lightning_lite.utilities.xla_device import XLADeviceUtils  # noqa: E402
 
 _TPU_AVAILABLE = XLADeviceUtils.tpu_device_exists()
 
 
@@ -16,7 +16,7 @@
 
 import pytest
 
-import pytorch_lightning.utilities.xla_device as xla_utils
+import lightning_lite.utilities.xla_device as xla_utils
 from pytorch_lightning.utilities import _XLA_AVAILABLE
 from tests_pytorch.helpers.runif import RunIf