fix xla launcher

Lightning-AI · Aug 28, 2022 · cd9bf08 · cd9bf08
1 parent 151755c
commit cd9bf08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pytorch_lightning/strategies/launchers/xla.py b/src/pytorch_lightning/strategies/launchers/xla.py
@@ -103,7 +103,7 @@ def _wrapping_function(
         return_queue: SimpleQueue,
         global_states: Optional[_GlobalStateSnapshot] = None,
     ) -> None:
-        self._local_rank = process_idx
+        self._strategy._local_rank = process_idx
         results = function(*args, **kwargs)
 
         if trainer is not None: