lightning/tests/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
test_kubeflow_environment.py Introduce `ClusterEnvironment.detect()` (#10564) 2021-11-19 12:24:10 +00:00
test_lightning_environment.py Introduce `ClusterEnvironment.detect()` (#10564) 2021-11-19 12:24:10 +00:00
test_lsf_environment.py Modify LSFEnvironment to use more reliable environment variable (#10825) 2022-01-05 12:45:25 +00:00
test_slurm_environment.py Add job_name as a staticmethod in SLURMEnvironment class (#10698) 2021-12-01 00:01:44 +00:00
test_torchelastic_environment.py Refactor `TorchElasticEnvironment.detect` to use `torch.distributed.is_torchelastic_launched` (#12376) 2022-03-21 16:51:24 +01:00
torch_elastic_deadlock.py Update `tests/plugins/*.py` to use `devices` instead of `gpus` or `ipus` (#11872) 2022-02-21 22:57:21 +01:00