lightning/tests/accelerators
Sean Naren bac8b1be81
Add support for CPU AMP autocast (#9084)
2021-08-25 12:18:00 +00:00
..
__init__.py Automatically find and run special tests (#6669) 2021-03-26 17:04:59 +00:00
ddp_model.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_accelerator_connector.py Add validate logic for precision (#9080) 2021-08-24 20:00:09 +00:00
test_common.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_cpu.py Add support for CPU AMP autocast (#9084) 2021-08-25 12:18:00 +00:00
test_ddp.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_ddp_spawn.py Deprecate LightningModule.summarize() in favor of pl.utilities.model_summary.summarize() (#8513) 2021-08-03 22:08:51 +00:00
test_dp.py Deprecate LightningModule.summarize() in favor of pl.utilities.model_summary.summarize() (#8513) 2021-08-03 22:08:51 +00:00
test_ipu.py Replace DataLoader sampler once for IPUs (#8858) 2021-08-16 11:28:05 +02:00
test_multi_nodes_gpu.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_tpu_backend.py Raise exception for ddp_cpu not supported for TPUs (#8530) 2021-07-26 13:52:34 +00:00