lightning/tests/tests_fabric/utilities
Corwin Joy 631911c004
Add special logic for 'step' in _optimizer_to_device (#20019)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2024-08-05 17:17:06 -04:00
..
__init__.py Rename LightningLite to Fabric (#16244) 2023-01-04 10:57:18 -05:00
test_apply_func.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_cloud_io.py precommit: ruff-format (#19434) 2024-02-15 13:39:17 -05:00
test_consolidate_checkpoint.py Revise checkpoint consolidation with PyTorch 2.3 (#19561) 2024-03-04 10:13:31 -05:00
test_data.py Refactor BoringFabric in tests (#19364) 2024-01-30 23:32:45 +01:00
test_device_dtype_mixin.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_device_parser.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_distributed.py Fix parameter count in ModelSummary when parameters are DTensors (#20163) 2024-08-05 10:57:31 -04:00
test_imports.py ci: testing with py3.11 (#17822) 2023-08-28 15:15:12 +00:00
test_init.py Drop PyTorch 2.0 from the test matrix (#20009) 2024-06-30 18:02:00 -04:00
test_load.py Remove support for PyTorch 1.13 (#19706) 2024-04-27 01:24:07 -04:00
test_logger.py Sanitize argument-free object params before logging (#19771) 2024-06-06 14:51:48 -04:00
test_optimizer.py Add special logic for 'step' in _optimizer_to_device (#20019) 2024-08-05 17:17:06 -04:00
test_rank_zero.py PyTest random order for Fabric tests (#19040) 2023-11-22 16:41:49 -05:00
test_registry.py Remove support for Python 3.8 (#20071) 2024-07-12 10:33:35 -04:00
test_runif.py Compose RunIf utilities (#17520) 2023-05-05 01:21:58 +02:00
test_seed.py Avoid printing the seed info message multiple times (#20108) 2024-07-20 20:25:11 +02:00
test_spike.py Update tests for PyTorch 2.2.1 (#19521) 2024-02-23 13:11:34 -05:00
test_throughput.py Drop PyTorch 2.0 from the test matrix (#20009) 2024-06-30 18:02:00 -04:00
test_warnings.py precommit: ruff-format (#19434) 2024-02-15 13:39:17 -05:00