vcpkg/ports/libtorch/force-cuda-include.patch

13 lines
481 B
Diff
Raw Normal View History

2024-03-26 06:57:42 +08:00
diff --git a/cmake/public/cuda.cmake b/cmake/public/cuda.cmake
index e8e450d36c..8ba1778b4f 100644
--- a/cmake/public/cuda.cmake
+++ b/cmake/public/cuda.cmake
@@ -60,6 +60,7 @@ find_package(CUDAToolkit REQUIRED)
cmake_policy(POP)
+set(CUDAToolkit_INCLUDE_DIR "${CUDA_INCLUDE_DIRS}")
if(NOT CMAKE_CUDA_COMPILER_VERSION STREQUAL CUDAToolkit_VERSION OR
NOT CUDA_INCLUDE_DIRS STREQUAL CUDAToolkit_INCLUDE_DIR)
message(FATAL_ERROR "Found two conflicting CUDA installs:\n"