lightning/tests/tests_pytorch/strategies
Liyang90 7668a6bf59
Flexible and easy to use HSDP setting (#19504)
Co-authored-by: awaelchli <aedu.waelchli@gmail.com>
2024-06-05 20:15:03 -04:00
..
launchers Avoid interactions through test artifacts (#19821) 2024-04-28 11:56:40 -04:00
scripts precommit: ruff-format (#19434) 2024-02-15 13:39:17 -05:00
__init__.py
test_common.py Remove support for PyTorch 1.13 (#19706) 2024-04-27 01:24:07 -04:00
test_custom_strategy.py test: replacing `tmpdir` with `tmp_path` in `tests_pytorch/strategies` (#19651) 2024-03-17 01:55:38 -04:00
test_ddp.py Remove support for PyTorch 1.13 (#19706) 2024-04-27 01:24:07 -04:00
test_ddp_integration.py Avoid interactions through test artifacts (#19821) 2024-04-28 11:56:40 -04:00
test_ddp_integration_comm_hook.py test: replacing `tmpdir` with `tmp_path` in `tests_pytorch/strategies` (#19651) 2024-03-17 01:55:38 -04:00
test_deepspeed.py
test_fsdp.py Flexible and easy to use HSDP setting (#19504) 2024-06-05 20:15:03 -04:00
test_model_parallel.py (9/n) Support 2D Parallelism - Remaining Checkpoint Logic (#19888) 2024-05-22 18:13:41 -04:00
test_model_parallel_integration.py (9/n) Support 2D Parallelism - Remaining Checkpoint Logic (#19888) 2024-05-22 18:13:41 -04:00
test_registry.py Avoid interactions through test artifacts (#19821) 2024-04-28 11:56:40 -04:00
test_single_device.py precommit: ruff-format (#19434) 2024-02-15 13:39:17 -05:00
test_xla.py Drop XLA XRT support (#19232) 2024-01-10 18:39:20 +01:00