lightning/tests/plugins
Adrian Wälchli bf54ac1cad
fix NCCL error with non-consecutive trainer gpus (#8165)
* device ids in barrier


x


x


s


same fix for spawn


fix non-nccl 


x

* add changelog

* get nccl backend

* get backend

Co-authored-by: Kaushik B <45285388+kaushikb11@users.noreply.github.com>
2021-06-28 22:08:10 +02:00
..
environments Add kubeflow cluster environment (#7300) 2021-05-17 09:05:24 +01:00
__init__.py enable ddp as a plugin (#4285) 2020-10-22 05:15:51 -04:00
test_amp_plugins.py Make optimizers skippable when using amp (#7975) 2021-06-16 00:23:30 +00:00
test_cluster_integration.py Nuke RPC (#8101) 2021-06-23 18:31:13 +00:00
test_custom_plugin.py Add typings for evaluation_loop.py and remove some dead code (#7015) 2021-04-15 07:36:04 +00:00
test_ddp_fully_sharded_with_full_state_dict.py FSDP with full state dict (#7487) 2021-05-24 08:11:45 +01:00
test_ddp_plugin.py fix NCCL error with non-consecutive trainer gpus (#8165) 2021-06-28 22:08:10 +02:00
test_ddp_plugin_with_comm_hook.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
test_ddp_spawn_plugin.py Add `add_to_queue`/`get_from_queue` for DDP spawn(#7916) 2021-06-23 03:19:37 +02:00
test_deepspeed_plugin.py Fix Special Tests (#7841) 2021-06-16 19:39:03 +02:00
test_double_plugin.py Fix double precision + ddp_spawn (#6924) 2021-06-01 15:21:17 +00:00
test_plugins_registry.py Add `tpu_spawn_debug` to plugin registry (#7933) 2021-06-15 22:32:51 +00:00
test_sharded_plugin.py Fix Special Tests (#7841) 2021-06-16 19:39:03 +02:00
test_single_device_plugin.py Merge pull request #7872 from PyTorchLightning/refactor/logger-poc-changes 2021-06-08 09:04:16 -04:00
test_tpu_spawn.py Standardize positional datamodule and argument names (#7431) 2021-06-15 11:50:13 +00:00