lightning/tests/tests_fabric/plugins/precision
awaelchli 1a59097ab2
Drop support for PyTorch 1.12 (#19300)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
2024-01-26 11:44:24 -05:00
..
__init__.py Rename LightningLite to Fabric (#16244) 2023-01-04 10:57:18 -05:00
test_all.py Split `Precision.init_context` (#18734) 2023-10-09 12:34:30 -04:00
test_amp.py Remove outdated workaround for PyTorch autocast bug (#18634) 2023-09-29 08:33:43 -04:00
test_amp_integration.py Drop support for PyTorch 1.12 (#19300) 2024-01-26 11:44:24 -05:00
test_bitsandbytes.py Support 4bit BNB layers meta-device materialization (#19150) 2023-12-20 22:13:18 +01:00
test_deepspeed.py Split `Precision.init_context` (#18734) 2023-10-09 12:34:30 -04:00
test_deepspeed_integration.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_double.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_double_integration.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_fsdp.py Drop support for PyTorch 1.11 (#18691) 2023-10-04 20:30:44 +02:00
test_half.py Split `Precision.init_context` (#18734) 2023-10-09 12:34:30 -04:00
test_transformer_engine.py TransformerEngine fallback compute dtype (#19082) 2023-12-14 03:02:09 +01:00
test_utils.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_xla.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00
test_xla_integration.py ruff: replace isort with ruff +TPU (#17684) 2023-09-26 11:54:55 -04:00