diff --git a/Jenkinsfile b/Jenkinsfile index b677ccd2b78d..b46632a960d6 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -88,6 +88,7 @@ stage('Build') { mkdir -p build cd build cp ../cmake/config.cmake . + echo set\\(USE_CUBLAS ON\\) >> config.cmake echo set\\(USE_CUDNN ON\\) >> config.cmake echo set\\(USE_CUDA ON\\) >> config.cmake echo set\\(USE_OPENGL ON\\) >> config.cmake diff --git a/cmake/util/FindCUDA.cmake b/cmake/util/FindCUDA.cmake index 41936acca217..83d2e6500fea 100644 --- a/cmake/util/FindCUDA.cmake +++ b/cmake/util/FindCUDA.cmake @@ -50,6 +50,9 @@ macro(find_cuda use_cuda) find_library(CUDA_CUDNN_LIBRARY cudnn ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64 ${CUDA_TOOLKIT_ROOT_DIR}/lib/win32) + find_library(CUDA_CUBLAS_LIBRARY cublas + ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64 + ${CUDA_TOOLKIT_ROOT_DIR}/lib/win32) else(MSVC) find_library(_CUDA_CUDA_LIBRARY cuda PATHS ${CUDA_TOOLKIT_ROOT_DIR} @@ -63,6 +66,9 @@ macro(find_cuda use_cuda) find_library(CUDA_CUDNN_LIBRARY cudnn ${CUDA_TOOLKIT_ROOT_DIR}/lib64 ${CUDA_TOOLKIT_ROOT_DIR}/lib) + find_library(CUDA_CUBLAS_LIBRARY cublas + ${CUDA_TOOLKIT_ROOT_DIR}/lib64 + ${CUDA_TOOLKIT_ROOT_DIR}/lib) endif(MSVC) endif(CUDA_FOUND) endmacro(find_cuda)