amend computation of the root dir of CUDAToolkit: CUDAToolkit_INCLUDE…

…_DIR is not defined, but CUDAToolkit_LIBRARY_DIR is
ValeevGroup · Jun 24, 2024 · 0885420 · 0885420
1 parent 8483e7f
commit 0885420
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/external/cuda.cmake b/external/cuda.cmake
@@ -28,7 +28,7 @@ foreach (library cublas;nvToolsExt)
 endforeach()
 
 if (NOT DEFINED CUDAToolkit_ROOT)
- get_filename_component(CUDAToolkit_ROOT "${CUDAToolkit_INCLUDE_DIR}/../" ABSOLUTE CACHE)
+ get_filename_component(CUDAToolkit_ROOT "${CUDAToolkit_LIBRARY_DIR}/../" ABSOLUTE CACHE)
 endif(NOT DEFINED CUDAToolkit_ROOT)
 
 # sanitize implicit dirs if CUDA host compiler != C++ compiler