From 9e6c26b75c836da355500768f27966711f94c3ce Mon Sep 17 00:00:00 2001 From: Robert Maynard Date: Wed, 12 Jan 2022 15:06:49 -0500 Subject: FindCUDAToolkit: depend cublas on cublasLt for CUDA 11.0+ --- Modules/FindCUDAToolkit.cmake | 7 +++++++ 1 file changed, 7 insertions(+) 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) -- cgit v0.12