lightning/tests/tests_fabric
Adrian Wälchli 080eaf38fa
Enable setting the sharding strategy as string in FSDP (#18087)
2023-07-15 18:07:09 +02:00
..
accelerators Show CUDA matmul precision info only ever once (#17960) 2023-07-04 03:47:27 -04:00
graveyard [TPU] Rename classes to use XLA instead of TPU (#17383) 2023-04-28 12:36:22 -04:00
helpers Remove automatic sharding support with `Fabric.run` or `fabric.launch(fn)` (#17832) 2023-06-15 16:02:09 +00:00
loggers Fix detection of next version in Fabric's CSVLogger (#17986) 2023-07-14 16:08:16 -04:00
plugins More XLA fixes for nightly support (#18085) 2023-07-15 01:16:42 +02:00
strategies Enable setting the sharding strategy as string in FSDP (#18087) 2023-07-15 18:07:09 +02:00
utilities Requirements update (#18014) 2023-07-10 13:00:20 +00:00
__init__.py Rename LightningLite to Fabric (#16244) 2023-01-04 10:57:18 -05:00
conftest.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
run_standalone_tests.sh Rename LightningLite to Fabric (#16244) 2023-01-04 10:57:18 -05:00
run_tpu_tests.sh replace local adjustment script with external (#17582) 2023-05-29 19:34:04 +00:00
test_cli.py ruff: autofix PT (#17541) 2023-05-04 11:50:39 -04:00
test_connector.py Support all half-precision modes in FSDP precision plugin (#17807) 2023-07-09 18:40:46 +00:00
test_fabric.py More XLA fixes for nightly support (#18085) 2023-07-15 01:16:42 +02:00
test_wrappers.py Reduce false positive warnings when calling module methods in Fabric (#17875) 2023-06-26 17:35:27 +02:00