Update find cublas so it search default path if needed. (#9149)

apache · Sep 30, 2021 · fafd164 · fafd164
1 parent da5b4fb
commit fafd164
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cmake/utils/FindCUDA.cmake b/cmake/utils/FindCUDA.cmake
@@ -89,12 +89,16 @@ macro(find_cuda use_cuda use_cudnn)
         ${CUDA_TOOLKIT_ROOT_DIR}/lib64
         ${CUDA_TOOLKIT_ROOT_DIR}/lib
         NO_DEFAULT_PATH)
+      # search default path if cannot find cublas in non-default
+      find_library(CUDA_CUBLAS_LIBRARY cublas)
       find_library(CUDA_CUBLASLT_LIBRARY
         NAMES cublaslt cublasLt
         PATHS
         ${CUDA_TOOLKIT_ROOT_DIR}/lib64
         ${CUDA_TOOLKIT_ROOT_DIR}/lib
         NO_DEFAULT_PATH)
+      # search default path if cannot find cublaslt in non-default
+      find_library(CUDA_CUBLASLT_LIBRARY NAMES cublaslt cublasLt)
     endif(MSVC)
 
     # find cuDNN