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 2023/04/22 12:46:21 UTC
[tvm] branch main updated: unify search path approach to various libs (#14694)
This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new d27837ac29 unify search path approach to various libs (#14694)
d27837ac29 is described below
commit d27837ac29fcf492266c4ed6b0931a2804498352
Author: Thomas Viehmann <tv...@beamnet.de>
AuthorDate: Sat Apr 22 14:46:14 2023 +0200
unify search path approach to various libs (#14694)
---
cmake/utils/FindCUDA.cmake | 19 +++++++++----------
1 file changed, 9 insertions(+), 10 deletions(-)
diff --git a/cmake/utils/FindCUDA.cmake b/cmake/utils/FindCUDA.cmake
index 607f1761ae..9ef8d33242 100644
--- a/cmake/utils/FindCUDA.cmake
+++ b/cmake/utils/FindCUDA.cmake
@@ -76,7 +76,7 @@ macro(find_cuda use_cuda use_cudnn)
else(MSVC)
find_library(_CUDA_CUDA_LIBRARY cuda
PATHS ${CUDA_TOOLKIT_ROOT_DIR}
- PATH_SUFFIXES lib lib64 targets/x86_64-linux/lib targets/x86_64-linux/lib/stubs lib64/stubs
+ PATH_SUFFIXES lib lib64 targets/x86_64-linux/lib targets/x86_64-linux/lib/stubs lib64/stubs lib/x86_64-linux-gnu
NO_DEFAULT_PATH)
if(_CUDA_CUDA_LIBRARY)
set(CUDA_CUDA_LIBRARY ${_CUDA_CUDA_LIBRARY})
@@ -86,20 +86,19 @@ macro(find_cuda use_cuda use_cudnn)
PATH_SUFFIXES lib lib64 targets/x86_64-linux/lib targets/x86_64-linux/lib/stubs lib64/stubs lib/x86_64-linux-gnu
NO_DEFAULT_PATH)
find_library(CUDA_CURAND_LIBRARY curand
- ${CUDA_TOOLKIT_ROOT_DIR}/lib64
- ${CUDA_TOOLKIT_ROOT_DIR}/lib
+ PATHS ${CUDA_TOOLKIT_ROOT_DIR}
+ PATH_SUFFIXES lib lib64 targets/x86_64-linux/lib targets/x86_64-linux/lib/stubs lib64/stubs lib/x86_64-linux-gnu
NO_DEFAULT_PATH)
find_library(CUDA_CUBLAS_LIBRARY cublas
- ${CUDA_TOOLKIT_ROOT_DIR}/lib64
- ${CUDA_TOOLKIT_ROOT_DIR}/lib
+ PATHS ${CUDA_TOOLKIT_ROOT_DIR}
+ PATH_SUFFIXES lib lib64 targets/x86_64-linux/lib targets/x86_64-linux/lib/stubs lib64/stubs lib/x86_64-linux-gnu
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
+ PATHS ${CUDA_TOOLKIT_ROOT_DIR}
+ PATH_SUFFIXES lib lib64 targets/x86_64-linux/lib targets/x86_64-linux/lib/stubs lib64/stubs lib/x86_64-linux-gnu
NO_DEFAULT_PATH)
# search default path if cannot find cublaslt in non-default
find_library(CUDA_CUBLASLT_LIBRARY NAMES cublaslt cublasLt)
@@ -115,8 +114,8 @@ macro(find_cuda use_cuda use_cudnn)
${CUDA_TOOLKIT_ROOT_DIR}/lib/Win32)
else(MSVC)
find_library(CUDA_CUDNN_LIBRARY cudnn
- ${CUDA_TOOLKIT_ROOT_DIR}/lib64
- ${CUDA_TOOLKIT_ROOT_DIR}/lib
+ PATHS ${CUDA_TOOLKIT_ROOT_DIR}
+ PATH_SUFFIXES lib lib64 targets/x86_64-linux/lib targets/x86_64-linux/lib/stubs lib64/stubs lib/x86_64-linux-gnu
NO_DEFAULT_PATH)
# search default path if cannot find cudnn in non-default
find_library(CUDA_CUDNN_LIBRARY cudnn)