lightning/tests/tests_pytorch/strategies
Carlos Mocholí 0a928e8ead
Reuse code in demos.BoringModel (#16242)
2023-01-06 09:47:08 +01:00
..
launchers Rename LightningLite to Fabric (#16244) 2023-01-04 10:57:18 -05:00
scripts Remove the deprecated `pl.utilities.cli` module (#16116) 2022-12-19 23:24:25 +01:00
__init__.py
test_bagua_strategy.py Skip a failing Bagua test for manual optimization (#16225) 2023-01-03 14:31:30 +00:00
test_colossalai.py Add support for colossalai 0.1.11 (#15888) 2022-12-20 05:57:03 +01:00
test_common.py Use sklearn in runif (#15426) 2022-11-01 11:40:32 +00:00
test_custom_plugin.py
test_ddp.py Add info message for Ampere GPUs to enable tf32 matmuls (#16037) 2022-12-13 19:24:31 +01:00
test_ddp_fully_sharded_native.py Rename LightningLite to Fabric (#16244) 2023-01-04 10:57:18 -05:00
test_ddp_fully_sharded_with_full_state_dict.py Update CI to CUDA 11.7.1 (#16123) 2023-01-04 22:44:23 +00:00
test_ddp_spawn.py Use sklearn in runif (#15426) 2022-11-01 11:40:32 +00:00
test_ddp_spawn_strategy.py Fix result transfer in multiprocessing launcher on multi-node (#15567) 2022-11-08 13:07:58 +01:00
test_ddp_strategy.py Rename LightningLite to Fabric (#16244) 2023-01-04 10:57:18 -05:00
test_ddp_strategy_with_comm_hook.py Remove checks for torch greater than 1.10 (#15846) 2022-12-01 21:41:22 +01:00
test_deepspeed_strategy.py Reuse code in demos.BoringModel (#16242) 2023-01-06 09:47:08 +01:00
test_dp.py Use sklearn in runif (#15426) 2022-11-01 11:40:32 +00:00
test_hivemind.py simplify torch.Tensor (#16190) 2022-12-24 06:44:27 +00:00
test_registry.py Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00
test_sharded_strategy.py Reuse code in demos.BoringModel (#16242) 2023-01-06 09:47:08 +01:00
test_single_device_strategy.py Merge pull request #13123 from Lightning-AI/mps_accelerator 2022-06-24 08:15:48 -04:00
test_tpu_spawn.py Refactor XLA and TPU checks across codebase (#14550) 2022-10-04 22:54:14 +00:00