2024-03-26 06:57:42 +08:00
|
|
|
diff --git a/cmake/public/cuda.cmake b/cmake/public/cuda.cmake
|
2024-07-04 11:01:35 +08:00
|
|
|
index 5964429..d528597 100644
|
2024-03-26 06:57:42 +08:00
|
|
|
--- a/cmake/public/cuda.cmake
|
|
|
|
|
+++ b/cmake/public/cuda.cmake
|
2024-07-04 11:01:35 +08:00
|
|
|
@@ -57,6 +57,7 @@ find_package(CUDAToolkit REQUIRED)
|
2024-03-26 06:57:42 +08:00
|
|
|
|
|
|
|
|
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"
|
