We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 368c71c commit fa9da18Copy full SHA for fa9da18
src/lightning/pytorch/strategies/launchers/multiprocessing.py
@@ -35,10 +35,10 @@
35
from lightning.pytorch.trainer.connectors.signal_connector import _SIGNUM
36
from lightning.pytorch.trainer.states import TrainerFn, TrainerState
37
from lightning.pytorch.utilities.rank_zero import rank_zero_debug
38
-from lightning.fabric.utilities.imports import _LIGHTNING_XPU_AVAILABLE
+from lightning.pytorch.utilities.imports import _LIGHTNING_XPU_AVAILABLE
39
40
if _LIGHTNING_XPU_AVAILABLE:
41
- from lightning_xpu.fabric import XPUAccelerator
+ from lightning_xpu.pytorch import XPUAccelerator
42
43
log = logging.getLogger(__name__)
44
0 commit comments