lightning/pytorch_lightning/accelerators
Rohit Gupta 475d7d0bd6
Add cpu device parser to validate cpu devices (#12160)
2022-03-28 20:33:04 +05:30
..
__init__.py Add support for Habana accelerator (HPU) (#11808) 2022-03-25 10:24:52 +00:00
accelerator.py Add support for Habana accelerator (HPU) (#11808) 2022-03-25 10:24:52 +00:00
cpu.py Add cpu device parser to validate cpu devices (#12160) 2022-03-28 20:33:04 +05:30
gpu.py Drop PyTorch 1.7 support (#12432) 2022-03-27 21:31:20 +00:00
hpu.py Add support for Habana accelerator (HPU) (#11808) 2022-03-25 10:24:52 +00:00
ipu.py Add `AcceleratorRegistry` (#12180) 2022-03-24 18:29:32 +00:00
registry.py Add `AcceleratorRegistry` (#12180) 2022-03-24 18:29:32 +00:00
tpu.py Add `AcceleratorRegistry` (#12180) 2022-03-24 18:29:32 +00:00