lightning/tests/tests_pytorch/accelerators
HMellor 07b39c257b
Cast on host instead of IPU when using `precision=16` (#13880)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
2022-07-28 19:26:41 +00:00
..
__init__.py Future 4/n: test & legacy in test/ folder (#13295) 2022-06-15 18:10:49 -04:00
test_accelerator_connector.py Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00
test_accelerator_registry.py Merge different gpu backends with accelerator='gpu' (#13642) 2022-07-25 14:46:45 +00: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 Add auto_device_count and device name support (#13423) 2022-07-22 10:29:02 +05:30
test_ipu.py Cast on host instead of IPU when using `precision=16` (#13880) 2022-07-28 19:26:41 +00:00
test_mps.py Merge pull request #13123 from Lightning-AI/mps_accelerator 2022-06-24 08:15:48 -04:00
test_tpu.py Fix TPU testing and collect all tests (#11098) 2022-07-27 15:40:40 +00:00