lightning/tests/tests_fabric/strategies
awaelchli 7d1a70752f
Update PyTorch 2.4 tests (#20079)
2024-07-13 05:09:09 -04:00
..
launchers Update tests for PyTorch 2.2.1 (#19521) 2024-02-23 13:11:34 -05:00
__init__.py
test_ddp.py Remove support for PyTorch 1.13 (#19706) 2024-04-27 01:24:07 -04:00
test_ddp_integration.py Drop PyTorch 2.0 from the test matrix (#20009) 2024-06-30 18:02:00 -04:00
test_deepspeed.py `set_device` before `init_process_group` (#19184) 2023-12-21 16:28:16 +01:00
test_deepspeed_integration.py Set `weights_only` in tests to avoid warnings in PyTorch 2.4 (#20057) 2024-07-08 04:38:27 -04:00
test_dp.py Add testing for PyTorch 2.4 (Fabric) (#20028) 2024-07-02 18:01:03 -04:00
test_fsdp.py Drop PyTorch 2.0 from the test matrix (#20009) 2024-06-30 18:02:00 -04:00
test_fsdp_integration.py Set `weights_only` in tests to avoid warnings in PyTorch 2.4 (#20057) 2024-07-08 04:38:27 -04:00
test_model_parallel.py Add testing for PyTorch 2.4 (Fabric) (#20028) 2024-07-02 18:01:03 -04:00
test_model_parallel_integration.py Update PyTorch 2.4 tests (#20079) 2024-07-13 05:09:09 -04:00
test_registry.py Support shortcut name for DeepSpeed stage 1 offload (#19075) 2024-02-05 20:53:18 -05:00
test_single_device.py Rewrite gradient clipping tests (#19262) 2024-01-10 12:39:56 -05:00
test_strategy.py Remove support for PyTorch 1.13 (#19706) 2024-04-27 01:24:07 -04:00
test_xla.py Refactor BoringFabric in tests (#19364) 2024-01-30 23:32:45 +01:00
test_xla_fsdp.py Remove support for PyTorch 1.13 (#19706) 2024-04-27 01:24:07 -04:00
test_xla_fsdp_integration.py Remove support for PyTorch 1.13 (#19706) 2024-04-27 01:24:07 -04:00