Skip to content

Commit fa9da18

Browse files
authored
change to lightning.pytorch
1 parent 368c71c commit fa9da18

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/lightning/pytorch/strategies/launchers/multiprocessing.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,10 +35,10 @@
3535
from lightning.pytorch.trainer.connectors.signal_connector import _SIGNUM
3636
from lightning.pytorch.trainer.states import TrainerFn, TrainerState
3737
from lightning.pytorch.utilities.rank_zero import rank_zero_debug
38-
from lightning.fabric.utilities.imports import _LIGHTNING_XPU_AVAILABLE
38+
from lightning.pytorch.utilities.imports import _LIGHTNING_XPU_AVAILABLE
3939

4040
if _LIGHTNING_XPU_AVAILABLE:
41-
from lightning_xpu.fabric import XPUAccelerator
41+
from lightning_xpu.pytorch import XPUAccelerator
4242

4343
log = logging.getLogger(__name__)
4444

0 commit comments

Comments
 (0)