lightning/tests/tests_pytorch/accelerators
Rohit Gupta f5bd6e6f5f
Cast only floating types with IPUs (#13983)
2022-08-04 19:46:07 +00:00
..
__init__.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_common.py Add support for DDP fork (#13405) 2022-07-22 16:05:35 +00:00
test_cpu.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_gpu.py Rename GPUAccelerator to CUDAAccelerator 2022-07-19 13:06:30 -04:00
test_hpu.py Added support for HPU device stats monitor (#13819) 2022-08-02 13:31:31 +05:30
test_ipu.py Cast only floating types with IPUs (#13983) 2022-08-04 19:46:07 +00:00
test_mps.py Merge pull request #13123 from Lightning-AI/mps_accelerator 2022-06-24 08:15:48 -04:00
test_registry.py Organize accelerator tests (#13986) 2022-08-03 13:49:55 +00:00
test_tpu.py Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00