You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by tq...@apache.org on 2020/10/18 16:16:56 UTC
[incubator-tvm] branch ci-docker-staging updated: Fix cuda build
This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch ci-docker-staging
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git
The following commit(s) were added to refs/heads/ci-docker-staging by this push:
new bf826ea Fix cuda build
bf826ea is described below
commit bf826ea6f2afaf368600d5aacd00b785ae9d77d7
Author: tqchen <tq...@octoml.ai>
AuthorDate: Sun Oct 18 09:16:37 2020 -0700
Fix cuda build
---
cmake/utils/FindCUDA.cmake | 11 ++++++++---
1 file changed, 8 insertions(+), 3 deletions(-)
diff --git a/cmake/utils/FindCUDA.cmake b/cmake/utils/FindCUDA.cmake
index e651751..564b837 100644
--- a/cmake/utils/FindCUDA.cmake
+++ b/cmake/utils/FindCUDA.cmake
@@ -87,15 +87,20 @@ macro(find_cuda use_cuda)
NO_DEFAULT_PATH)
find_library(CUDA_CUDNN_LIBRARY cudnn
${CUDA_TOOLKIT_ROOT_DIR}/lib64
- ${CUDA_TOOLKIT_ROOT_DIR}/lib)
+ ${CUDA_TOOLKIT_ROOT_DIR}/lib
+ NO_DEFAULT_PATH)
+ # search default path if cannot find cudnn in non-default
+ find_library(CUDA_CUDNN_LIBRARY cudnn)
find_library(CUDA_CUBLAS_LIBRARY cublas
${CUDA_TOOLKIT_ROOT_DIR}/lib64
- ${CUDA_TOOLKIT_ROOT_DIR}/lib)
+ ${CUDA_TOOLKIT_ROOT_DIR}/lib
+ NO_DEFAULT_PATH)
find_library(CUDA_CUBLASLT_LIBRARY
NAMES cublaslt cublasLt
PATHS
${CUDA_TOOLKIT_ROOT_DIR}/lib64
- ${CUDA_TOOLKIT_ROOT_DIR}/lib)
+ ${CUDA_TOOLKIT_ROOT_DIR}/lib
+ NO_DEFAULT_PATH)
endif(MSVC)
message(STATUS "Found CUDA_TOOLKIT_ROOT_DIR=" ${CUDA_TOOLKIT_ROOT_DIR})
message(STATUS "Found CUDA_CUDA_LIBRARY=" ${CUDA_CUDA_LIBRARY})