lightning/tests/plugins
Rohit Gupta 02dc7751e9
Add `bfloat16` support to `DeepSpeedStrategy` (#12508)
Co-authored-by: carmocca <carlossmocholi@gmail.com>
2022-04-07 22:18:51 +00:00
..
environments Clear env before test cases requiring empty env (#12561) 2022-04-07 15:44:28 +00:00
precision Add `bfloat16` support to `DeepSpeedStrategy` (#12508) 2022-04-07 22:18:51 +00:00
__init__.py enable ddp as a plugin (#4285) 2020-10-22 05:15:51 -04:00
test_amp_plugins.py Update `tests/plugins/*.py` to use `devices` instead of `gpus` or `ipus` (#11872) 2022-02-21 22:57:21 +01:00
test_checkpoint_io_plugin.py Integrate global step with progress tracking (#11805) 2022-03-07 19:21:37 +00:00
test_cluster_integration.py Refactor `TorchElasticEnvironment.detect` to use `torch.distributed.is_torchelastic_launched` (#12376) 2022-03-21 16:51:24 +01:00
test_double_plugin.py Update `tests/plugins/*.py` to use `devices` instead of `gpus` or `ipus` (#11872) 2022-02-21 22:57:21 +01:00