lightning/tests/accelerators
Ning f6ed0bd8ca
introduce has_len_all_ranks() to check the length of dataloader across ranks (#9827)
* introduce , udpate tests

* update CHANGELOG.md

* change staticmethod and hook attribute naming

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* fix typo

* remove non-essential comment

* fix merge error and comment format

* try to fix test_tpu.py failure

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* update on comments

* chlog

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* chlog

* update

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* try fix

* Revert back TPUSpawn changes

* Update test

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: thomas chaton <thomas@grid.ai>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: Kaushik B <kaushikbokka@gmail.com>
2021-11-02 13:22:58 -04:00
..
__init__.py Automatically find and run special tests (#6669) 2021-03-26 17:04:59 +00:00
ddp_model.py Update strategy flag in docs (#10000) 2021-10-20 21:02:53 +05:30
test_accelerator_connector.py Fix `distrib_type` not being set when Plugin instances being passed to Trainer (#10251) 2021-11-01 17:11:57 +05:30
test_common.py Add `auto_device_count` method to `Accelerators` (#10222) 2021-10-29 22:31:32 +02:00
test_cpu.py Unify checkpoint load paths [redo #9693] (#10061) 2021-10-25 19:05:31 +00:00
test_ddp.py Update strategy flag in docs (#10000) 2021-10-20 21:02:53 +05:30
test_ddp_spawn.py rename set_random_master_port (#10104) 2021-10-25 12:09:05 +00:00
test_dp.py rename set_random_master_port (#10104) 2021-10-25 12:09:05 +00:00
test_gpu.py [1/4] Add get_device_stats to accelerator interface (#9586) 2021-09-26 21:09:16 -07:00
test_ipu.py Add support for `devices='auto'` (#10264) 2021-10-30 15:05:51 +00:00
test_multi_nodes_gpu.py (1/n) tests: Use strategy flag instead of accelerator for training strategies (#9931) 2021-10-16 20:40:25 +05:30
test_tpu.py introduce has_len_all_ranks() to check the length of dataloader across ranks (#9827) 2021-11-02 13:22:58 -04:00