From dec9b9f01de72bf5c2a47bfe5ce2161b8fa2cfb6 Mon Sep 17 00:00:00 2001 From: awaelchli Date: Tue, 29 Aug 2023 12:22:09 +0200 Subject: [PATCH] Fix incomplete cherry-pick for #16966 --- src/lightning/fabric/strategies/xla.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/src/lightning/fabric/strategies/xla.py b/src/lightning/fabric/strategies/xla.py index ba493401026b9..da2326e473778 100644 --- 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):