xla
c48478aa - Remove CUDA tests from distributed tests. (#9612)

Commit
182 days ago
Remove CUDA tests from distributed tests. (#9612) This PR removes CUDA specific logic and tests from distributed tests. Including both multiprocessing and SPMD tests. This is in line with the CUDA deprecation that started on release 2.8. **Key Changes:** - Removed `skipIf` test decorations whenever the condition is checking CUDA - Removed `CUDA` from the list of allowed devices for a few of these tests
Author
Parents
Loading