lightning/tests/tests_pytorch/utilities
awaelchli 75d5a2d046 move pl/utilities/xla_device.py to lite/utilities/xla_device.py (#14514) 2022-09-05 17:36:02 +02:00
..
__init__.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_all_gather_grad.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_apply_func.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_argparse.py Remove the legacy `get_deprecated_arg_names` (#14415) 2022-08-29 14:53:57 +02:00
test_auto_restart.py fix imports of collections.abc for py3.10 (#14345) 2022-08-23 11:52:58 -04:00
test_cloud_io.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_data.py Allowed setting attributes on `DataLoader` and `BatchSampler` when instantiated inside `*_dataloader` hooks (#14212) 2022-08-17 11:42:54 -04:00
test_deepspeed_collate_checkpoint.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_deepspeed_model_summary.py Organize model summary utilities (#13893) 2022-07-28 19:23:29 +02:00
test_device_parser.py Add support for DDP fork (#13405) 2022-07-22 16:05:35 +00:00
test_distributed.py Update gather_all_tensors to handle tensors of different sizes (#12630) 2022-06-29 17:03:00 +00:00
test_dtype_device_mixin.py Move test_dtype_device_mixin to lite (#14511) 2022-09-05 16:31:00 +02:00
test_enums.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_fetching.py Remove mps config for test (#14379) 2022-08-26 02:47:37 -04:00
test_finite_checks.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_grads.py Fix dtype inference during gradient norm computation (#14051) 2022-08-08 11:35:06 +00:00
test_imports.py Clean up fairscale imports (#14476) 2022-09-01 18:08:40 +02:00
test_logger.py save checkpoints and profiler output to the first logger (#14325) 2022-08-26 17:23:54 +00:00
test_memory.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_migration.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_model_helpers.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_model_summary.py Organize model summary utilities (#13893) 2022-07-28 19:23:29 +02:00
test_optimizer.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_parameter_tying.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_parsing.py Fix `LightningDataModule` hparams parsing (#12806) 2022-08-25 18:57:48 +00:00
test_rank_zero.py Unify rank zero messaging utilities (#14116) 2022-08-30 09:51:30 +00:00
test_remote_filesystem.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_seed.py update rng state save/load test to also run on cuda gpu (#14396) 2022-08-31 16:36:35 +00:00
test_torchdistx.py Remove meta device utilities in favor of torchdistx (#13868) 2022-08-05 12:20:27 +00:00
test_types.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_upgrade_checkpoint.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_warnings.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00