lightning/tests/accelerators
Carlos Mocholí ba036fdeea
Support special test parametrizations (#10569)
2021-11-17 15:46:14 +00: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 Mock GPU accelerator connector tests (#10554) 2021-11-16 16:13:40 +00:00
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 Support special test parametrizations (#10569) 2021-11-17 15:46:14 +00:00
test_ddp_spawn.py Remove deprecated dataloader arguments in Trainer methods (#10325) 2021-11-04 11:03:39 +01:00
test_dp.py remove deprecated signature for `transfer_batch_to_device` (#10480) 2021-11-13 19:32:30 +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 Update recommendation on `dataloader_idx` (#10318) 2021-11-04 01:39:55 +01: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