lightning/tests/accelerators
sisilmehta2000 d337374da7
[FSDP] Adding Native FSDP Strategy (#12447)
2022-05-05 12:48:29 +00:00
..
__init__.py Automatically find and run special tests (#6669) 2021-03-26 17:04:59 +00:00
test_accelerator_connector.py [FSDP] Adding Native FSDP Strategy (#12447) 2022-05-05 12:48:29 +00:00
test_accelerator_registry.py Add support for Habana accelerator (HPU) (#11808) 2022-03-25 10:24:52 +00:00
test_common.py Remove AcceleratorConnector.parallel_devices (#12075) 2022-03-25 01:45:40 +00:00
test_cpu.py Raise `MisconfigurationException` when the accelerator is available but… (#12708) 2022-04-11 23:31:17 +05:30
test_gpu.py Drop PyTorch 1.7 support (#12432) 2022-03-27 21:31:20 +00:00
test_hpu.py Add support for Habana accelerator (HPU) (#11808) 2022-03-25 10:24:52 +00:00
test_ipu.py Deprecate `num_processes`,`gpus`, `tpu_cores`, and `ipus` from the Trainer constructor (#11040) 2022-04-10 17:10:05 +00:00
test_tpu.py Remove support for passing strategy strings to accelerator (#12696) 2022-04-11 22:52:28 +02:00