lightning/tests/accelerators
Kaushik B 041da417db
Remove TPU Availability check from parse devices (#12326)
* Remove TPU Availability check from parse devices
* Update tests
2022-03-29 15:57:43 +02:00
..
__init__.py
test_accelerator_connector.py Remove TPU Availability check from parse devices (#12326) 2022-03-29 15:57:43 +02: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 Add cpu device parser to validate cpu devices (#12160) 2022-03-28 20:33:04 +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 Update `tests/accelerators/*.py` to use `devices` instead of `gpus` or `ipus` (#11817) 2022-03-24 14:09:39 +00:00
test_tpu.py Remove `AcceleratorConnector.tpu_cores` and Deprecate `Trainer.tpu_cores` (#12437) 2022-03-29 04:37:22 +02:00