mirror/CMake
1
0
Fork 0
mirror of https://github.com/Kitware/CMake.git synced 2026-01-10 15:50:50 -06:00
Code Issues Packages Projects Releases Wiki Activity

FindCUDAToolkit: depend cublas on cublasLt for CUDA 11.0+

Browse Source
This commit is contained in:
Robert Maynard
2022-01-12 15:06:49 -05:00
parent 0c038689be
commit 9e6c26b75c
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
Modules/FindCUDAToolkit.cmake
View File

@@ -925,6 +925,13 @@ if(CUDAToolkit_FOUND)
_CUDAToolkit_find_and_add_import_lib(${cuda_lib}_static DEPS culibos)
endforeach()
if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 11.0.0)
# cublas depends on cublasLt
# https://docs.nvidia.com/cuda/archive/11.0/cublas/index.html#static-library
_CUDAToolkit_find_and_add_import_lib(cublas DEPS cublasLt)
_CUDAToolkit_find_and_add_import_lib(cublas_static DEPS cublasLt_static)
endif()
# cuFFTW depends on cuFFT
_CUDAToolkit_find_and_add_import_lib(cufftw DEPS cufft)
_CUDAToolkit_find_and_add_import_lib(cufftw DEPS cufft_static)