Commit 70f9a9f1 authored by Davis King's avatar Davis King

Made cusolver finding work in newer versions of cmake and cuda

parent f7f6f676
...@@ -716,10 +716,12 @@ if (NOT TARGET dlib) ...@@ -716,10 +716,12 @@ if (NOT TARGET dlib)
endif() endif()
# Find where cuSOLVER is since the FindCUDA cmake package doesn't # Find where cuSOLVER is since the FindCUDA cmake package doesn't
# bother to look for it. # bother to look for it in older versions of cmake.
get_filename_component(cuda_blas_path "${CUDA_CUBLAS_LIBRARIES}" DIRECTORY) if (NOT CUDA_cusolver_LIBRARY)
find_library(cusolver cusolver HINTS ${cuda_blas_path}) get_filename_component(cuda_blas_path "${CUDA_CUBLAS_LIBRARIES}" DIRECTORY)
mark_as_advanced(cusolver) find_library(CUDA_cusolver_LIBRARY cusolver HINTS ${cuda_blas_path})
mark_as_advanced(CUDA_cusolver_LIBRARY)
endif()
# Also find OpenMP since cuSOLVER needs it. Importantly, we only # Also find OpenMP since cuSOLVER needs it. Importantly, we only
# look for one to link to if our use of BLAS, specifically the # look for one to link to if our use of BLAS, specifically the
# Intel MKL, hasn't already decided what to use. This is because # Intel MKL, hasn't already decided what to use. This is because
...@@ -756,7 +758,7 @@ if (NOT TARGET dlib) ...@@ -756,7 +758,7 @@ if (NOT TARGET dlib)
${CUDA_CUBLAS_LIBRARIES} ${CUDA_CUBLAS_LIBRARIES}
${cudnn} ${cudnn}
${CUDA_curand_LIBRARY} ${CUDA_curand_LIBRARY}
${cusolver} ${CUDA_cusolver_LIBRARY}
) )
if(openmp_libraries) if(openmp_libraries)
list(APPEND dlib_needed_libraries ${openmp_libraries}) list(APPEND dlib_needed_libraries ${openmp_libraries})
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment