diff options
Diffstat (limited to 'vcpkg/ports/libtorch/fix-dist-cuda.patch')
| -rw-r--r-- | vcpkg/ports/libtorch/fix-dist-cuda.patch | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/vcpkg/ports/libtorch/fix-dist-cuda.patch b/vcpkg/ports/libtorch/fix-dist-cuda.patch new file mode 100644 index 0000000..65e4bc1 --- /dev/null +++ b/vcpkg/ports/libtorch/fix-dist-cuda.patch @@ -0,0 +1,22 @@ +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) |