lightning/pytorch_lightning/plugins/environments
Danielle Pintz caed77f155
Refactor `TorchElasticEnvironment.detect` to use `torch.distributed.is_torchelastic_launched` (#12376)
* Refactor TorchElasticEnvironment.detect to use native utility from torch.distributed

* fix version and tests

* fix version

* Update tests/accelerators/test_accelerator_connector.py

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2022-03-21 16:51:24 +01:00
..
__init__.py Add `Trainer(strategy="bagua")` (#11146) 2022-02-04 17:02:09 +00:00
bagua_environment.py Add `Trainer(strategy="bagua")` (#11146) 2022-02-04 17:02:09 +00:00
cluster_environment.py Introduce `ClusterEnvironment.detect()` (#10564) 2021-11-19 12:24:10 +00:00
kubeflow_environment.py Use `Set` operations in `Environment.detect` (#10673) 2021-11-22 23:03:47 +05:30
lightning_environment.py Centralize rank_zero_only utilities into their own module (#11747) 2022-02-07 08:09:55 +00:00
lsf_environment.py fix typos (#11937) 2022-02-16 17:27:51 -08:00
slurm_environment.py Fix typing in `pl.plugins.environments` (#10943) 2021-12-07 02:14:02 +00:00
torchelastic_environment.py Refactor `TorchElasticEnvironment.detect` to use `torch.distributed.is_torchelastic_launched` (#12376) 2022-03-21 16:51:24 +01:00