diff options
| author | Brad King <brad.king@kitware.com> | 2022-01-13 20:23:51 (GMT) |
|---|---|---|
| committer | Kitware Robot <kwrobot@kitware.com> | 2022-01-13 20:23:58 (GMT) |
| commit | 874ae8045a85f1667ed27891647c6aae1ba9a8ba (patch) | |
| tree | 113c2358c50a27615351d8637e013956cc50527d | |
| parent | 6baf3b27c968a08e55ada33a47b6e68c618668cf (diff) | |
| parent | 9e6c26b75c836da355500768f27966711f94c3ce (diff) | |
| download | CMake-874ae8045a85f1667ed27891647c6aae1ba9a8ba.zip CMake-874ae8045a85f1667ed27891647c6aae1ba9a8ba.tar.gz CMake-874ae8045a85f1667ed27891647c6aae1ba9a8ba.tar.bz2 | |
Merge topic 'cudatoolkit_correct_cublas_deps'
9e6c26b75c FindCUDAToolkit: depend cublas on cublasLt for CUDA 11.0+
Acked-by: Kitware Robot <kwrobot@kitware.com>
Reviewed-by: Raul Tambre <raul@tambre.ee>
Merge-request: !6857
| -rw-r--r-- | Modules/FindCUDAToolkit.cmake | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/Modules/FindCUDAToolkit.cmake b/Modules/FindCUDAToolkit.cmake index de5f376..b2445a4 100644 --- a/Modules/FindCUDAToolkit.cmake +++ b/Modules/FindCUDAToolkit.cmake @@ -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) |