Fix incomplete cherry-pick for #16966

Lightning-AI · Aug 30, 2023 · efa83da · efa83da
1 parent 4a6ab85
commit efa83da
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/src/lightning/fabric/strategies/xla.py b/src/lightning/fabric/strategies/xla.py
@@ -207,11 +207,6 @@ def remove_checkpoint(self, filepath: _PATH) -> None:
     def register_strategies(cls, strategy_registry: _StrategyRegistry) -> None:
         strategy_registry.register("xla", cls, description=cls.__class__.__name__)
 
-    def _set_world_ranks(self) -> None:
-        if self.cluster_environment is None:
-            return
-        rank_zero_only.rank = self.cluster_environment.global_rank()
-
     @staticmethod
     def _validate_dataloader(dataloader: object) -> None:
         if not has_len(dataloader):