From 5c6b0e0de7ec72597e3ddf691d671cfacea99e52 Mon Sep 17 00:00:00 2001
From: "Tatsuya.Nishiyama" <nishiyama.tatsuya@lab.ntt.co.jp>
Date: Thu, 14 Jun 2018 14:29:24 +0900
Subject: [PATCH] Fix missing cublas libraries

---
 Jenkinsfile               | 1 +
 cmake/util/FindCUDA.cmake | 6 ++++++
 2 files changed, 7 insertions(+)

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)