lightning/tests/tests_fabric
Carlos Mocholí 67459944ea
Avoid FSDP deprecations during save/load with newer torch versions (#19463)
* Avoid FSDP deprecations during save/load with newer torch versions

* Refactor

* Tests
2024-02-14 19:43:59 +01:00
..
accelerators `set_device` before `init_process_group` (#19184) 2023-12-21 16:28:16 +01:00
graveyard [TPU] Proper half-precision implementation for XLA (#18213) 2023-08-11 11:37:41 -04:00
helpers Refactor BoringFabric in tests (#19364) 2024-01-30 23:32:45 +01:00
loggers Fix `CSVLogger` trying to append to file from previous run in same version folder (#19446) 2024-02-13 13:59:04 -05:00
plugins Drop support for PyTorch 1.12 (#19300) 2024-01-26 11:44:24 -05:00
strategies Avoid FSDP deprecations during save/load with newer torch versions (#19463) 2024-02-14 19:43:59 +01:00
utilities Create barrier without timeout in `prepare_data()` (#19448) 2024-02-13 12:10:07 +01:00
__init__.py Ignore pytest cleanup warning (#19164) 2023-12-29 07:52:19 +01:00
conftest.py Remove the Graphcore IPU integration (#19405) 2024-02-12 16:16:02 -05:00
run_tpu_tests.sh Drop XLA XRT support (#19232) 2024-01-10 18:39:20 +01:00
test_cli.py Rename Lightning Fabric CLI (#19442) 2024-02-12 17:22:53 +01:00
test_connector.py Error message to inform bitsandbytes is only supported on CUDA (#19360) 2024-01-29 19:52:28 -05:00
test_fabric.py Create barrier without timeout in `prepare_data()` (#19448) 2024-02-13 12:10:07 +01:00
test_wrappers.py Reapply `torch.compile` in Fabric.setup() (#19280) 2024-01-23 21:17:41 -05:00