lightning/tests/tests_fabric
Tianshu Wang 601c060805
Add `convert_module` to FSDP (#20323)
* Add `convert_module` to FSDP

* Update ChangeLog

* make plugin type check more flexible (#20186)

Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Co-authored-by: Luca Antiga <luca.antiga@gmail.com>

* Make plugin type check more flexible (Fabric) (#20452)

* make plugin type check more flexible

* Change signature and make the equivalent changes to Fabric connector

---------

Co-authored-by: Jianing Yang <jed970610@gmail.com>
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>

* Pin setuptools for gpu builds

* Fix link in doc

---------

Co-authored-by: Luca Antiga <luca.antiga@gmail.com>
Co-authored-by: Jianing Yang <jed970610@gmail.com>
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Co-authored-by: Luca Antiga <luca@lightning.ai>
2024-12-11 00:20:23 +01:00
..
accelerators Replace `using_pjrt()` xla runtime `device_type()` check with in xla.py for `torch-xla>=2.5` (#20442) 2024-11-25 11:47:19 +01:00
graveyard
helpers bump python 3.9+ (#20413) 2024-11-25 09:20:17 +01:00
loggers Remove confusing warning "Missing logger folder" (#20109) 2024-07-20 20:24:38 +02:00
plugins Add `convert_module` to FSDP (#20323) 2024-12-11 00:20:23 +01:00
strategies FP8 + FSDP2 + torch.compile examples for PyTorch Lightning and Fabric (#20440) 2024-11-26 15:29:40 +01:00
utilities bump python 3.9+ (#20413) 2024-11-25 09:20:17 +01:00
__init__.py
conftest.py FP8 + FSDP2 + torch.compile examples for PyTorch Lightning and Fabric (#20440) 2024-11-26 15:29:40 +01:00
run_tpu_tests.sh
test_cli.py Add `ddp_find_unused_parameters_true` alias in Fabric's DDPStrategy (#20125) 2024-08-07 10:47:36 -04:00
test_connector.py bump python 3.9+ (#20413) 2024-11-25 09:20:17 +01:00
test_fabric.py bump python 3.9+ (#20413) 2024-11-25 09:20:17 +01:00
test_wrappers.py Add Python 3.12 to the CPU test matrix (#20078) 2024-07-13 06:07:35 -04:00