lightning/tests/tests_pytorch/strategies
Rohit Gupta b25275ccc2
Cast to fp16 before moving to device with deepspeed (#14000)
2022-08-05 22:15:15 +00:00
..
launchers Snapshot selected globals and restore them in spawned process (#13921) 2022-08-01 22:21:46 +00:00
__init__.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
ddp_model.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_bagua_strategy.py Add support for DDP fork (#13405) 2022-07-22 16:05:35 +00:00
test_common.py Merge pull request #13123 from Lightning-AI/mps_accelerator 2022-06-24 08:15:48 -04:00
test_custom_plugin.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_ddp.py Add support for DDP fork (#13405) 2022-07-22 16:05:35 +00:00
test_ddp_fully_sharded_native.py Run GPU tests with PyTorch 1.12 (#13716) 2022-07-28 19:37:57 +05:30
test_ddp_fully_sharded_with_full_state_dict.py Add support for DDP fork (#13405) 2022-07-22 16:05:35 +00:00
test_ddp_spawn.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_ddp_spawn_strategy.py Rename spawn-based launchers (#13743) 2022-07-23 11:48:15 -04:00
test_ddp_strategy.py Enable timeout for `DDPStrategy` (#13244) 2022-06-21 15:49:57 +02:00
test_ddp_strategy_with_comm_hook.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_deepspeed_strategy.py Cast to fp16 before moving to device with deepspeed (#14000) 2022-08-05 22:15:15 +00:00
test_dp.py Raise an error if batch transfer hooks are overridden with IPUAccelerator (#13961) 2022-08-04 12:04:42 +00:00
test_hivemind.py add testing PT 1.12 (#13386) 2022-07-15 19:41:23 +02:00
test_sharded_strategy.py Run GPU tests with PyTorch 1.12 (#13716) 2022-07-28 19:37:57 +05:30
test_single_device_strategy.py Merge pull request #13123 from Lightning-AI/mps_accelerator 2022-06-24 08:15:48 -04:00
test_strategy_registry.py Add ddp_notebook alias for ddp_fork (#13744) 2022-07-23 09:06:35 -04:00
test_tpu_spawn.py Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00