pytorch
1718b16d - [Caffe2] gcs_cuda_only is trivial if CUDA not available (#44578)

Commit
4 years ago
[Caffe2] gcs_cuda_only is trivial if CUDA not available (#44578) Summary: Make `gcs_cuda_only` and `gcs_gpu_only` return empty device lists if CUDA/GPU(CUDA or RocM) not available Pull Request resolved: https://github.com/pytorch/pytorch/pull/44578 Reviewed By: walterddr Differential Revision: D23664227 Pulled By: malfet fbshipit-source-id: 176b5d964c0b02b8379777cd9a38698c11818690
Author
Parents
Loading