diff --git a/cmake/Dependencies.cmake b/cmake/Dependencies.cmake index 9acaf75..4f44d3e 100644 --- a/cmake/Dependencies.cmake +++ b/cmake/Dependencies.cmake @@ -1205,6 +1205,8 @@ if(USE_DISTRIBUTED AND USE_TENSORPIPE) list(APPEND Caffe2_DEPENDENCY_LIBS tensorpipe) list(APPEND Caffe2_DEPENDENCY_LIBS nlohmann) if(USE_CUDA) + add_library(tensorpipe_cuda ALIAS unofficial::tensorpipe::tensorpipe_cuda) + list(APPEND Caffe2_CUDA_DEPENDENCY_LIBS tensorpipe_cuda) elseif(USE_ROCM) message(WARNING "TensorPipe doesn't yet support ROCm") @@ -1279,7 +1281,7 @@ if(USE_GLOO) endif() # Pick the right dependency depending on USE_CUDA list(APPEND Caffe2_DEPENDENCY_LIBS gloo) - if(USE_CUDA) + if(FALSE) list(APPEND Caffe2_CUDA_DEPENDENCY_LIBS gloo_cuda) elseif(USE_ROCM) list(APPEND Caffe2_HIP_DEPENDENCY_LIBS gloo_hip)