lightning/tests/accelerators
four4fish e646ca1d59
Remove `setup_optimizers_in_pre_dispatch` logic (#10906)
2021-12-03 15:05:08 +01:00
..
__init__.py
ddp_model.py Update strategy flag in docs (#10000) 2021-10-20 21:02:53 +05:30
test_accelerator_connector.py Rename special to standalone (#10779) 2021-11-26 17:13:14 +00:00
test_common.py Add `auto_device_count` method to `Accelerators` (#10222) 2021-10-29 22:31:32 +02:00
test_cpu.py Remove `setup_optimizers_in_pre_dispatch` logic (#10906) 2021-12-03 15:05:08 +01:00
test_ddp.py Rename special to standalone (#10779) 2021-11-26 17:13: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 Remove `setup_optimizers_in_pre_dispatch` logic (#10906) 2021-12-03 15:05:08 +01:00
test_multi_nodes_gpu.py Rename special to standalone (#10779) 2021-11-26 17:13:14 +00:00
test_tpu.py 1/n Move precision plugin into strategy - update reference (#10570) 2021-11-19 00:39:01 +00:00