lightning/tests/plugins
Danielle Pintz 3392215ef6
Fix broken `test_cpu_amp_precision_context_manager` (#9809)
* @RunIf(min_gpus=1)

* dtype -> fast_dtype
2021-10-04 12:14:13 +00:00
..
environments Skip reconciliate_processes if used within a cluster environment that creates processes externally (#9389) 2021-09-15 11:54:17 +01:00
__init__.py
test_amp_plugins.py Fix broken `test_cpu_amp_precision_context_manager` (#9809) 2021-10-04 12:14:13 +00:00
test_checkpoint_io_plugin.py Add remove_checkpoint to CheckpointIO plugin to simplify ModelCheckpo… (#9373) 2021-09-10 11:55:04 +01:00
test_cluster_integration.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_custom_plugin.py Add property to skip restoring optimizers and schedulers via plugin (#8644) 2021-07-31 10:08:10 +02:00
test_ddp_fully_sharded_with_full_state_dict.py Remove `_NATIVE_AMP_AVAILABLE` checks (#9747) 2021-09-29 15:34:26 +02:00
test_ddp_plugin.py Call TrainingTypePlugin collective functions directly instead of going through the Accelerator (#9677) 2021-09-27 14:52:57 +02:00
test_ddp_plugin_with_comm_hook.py Support post-localSGD in Lightning DDP plugin (#8967) 2021-08-26 08:24:49 +01:00
test_ddp_spawn_plugin.py Move add_to_queue/get_from_queue to DDPSpawnPlugin (#9118) 2021-09-10 20:58:02 +00:00
test_deepspeed_plugin.py Fix Deepspeed and lightning calling scheduler (#9788) 2021-10-01 14:35:44 +00:00
test_double_plugin.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_plugins_registry.py Add DeepSpeed Stage 1 + doc improvements for model parallel (#8974) 2021-08-18 19:40:19 +05:30
test_sharded_plugin.py Remove `_NATIVE_AMP_AVAILABLE` checks (#9747) 2021-09-29 15:34:26 +02:00
test_single_device_plugin.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00
test_tpu_spawn.py Replace `yapf` with `black` (#7783) 2021-07-26 13:37:35 +02:00