lightning/pytorch_lightning/accelerators
Eric Wiener 3f78c4ca7a
Track CPU stats with DeviceStatsMonitor (#11795)
Co-authored-by: ananthsub <ananth.subramaniam@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Kaushik B <kaushikbokka@gmail.com>
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
2022-05-10 10:57:38 +00:00
..
__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 Track CPU stats with DeviceStatsMonitor (#11795) 2022-05-10 10:57:38 +00:00
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