correct usage in multiprocessing

Lightning-AI · Oct 25, 2023 · 1b90607 · 1b90607
1 parent c5018fc
commit 1b90607
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/lightning/fabric/strategies/launchers/multiprocessing.py b/src/lightning/fabric/strategies/launchers/multiprocessing.py
@@ -35,7 +35,7 @@
     from lightning.fabric.strategies import ParallelStrategy
 
 if _lightning_xpu_available():
-    from lightning_xpu.fabric import XPUAccelerator
+    from lightning_xpu.fabric import XPUAccelerator, is_xpu_initialized
 
 
 class _MultiProcessingLauncher(_Launcher):
@@ -258,7 +258,7 @@ def _check_bad_xpu_fork() -> None:
     Lightning users.
 
     """
-    if not XPUAccelerator.is_xpu_initialized():
+    if not is_xpu_initialized():
         return
 
     message = (