diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake index b80c35a0e6a849a94ecbe1b4650af4998f2d88c3..a5d3d572181bcd7555d112961eac497e1195cfe3 100644 --- a/cmake/cuda.cmake +++ b/cmake/cuda.cmake @@ -180,13 +180,16 @@ function(add_cuda_static_lib alias cuda_lib_paths file_name) add_library(${alias} STATIC IMPORTED GLOBAL) set_property(TARGET ${alias} PROPERTY IMPORTED_LOCATION ${ABS_PATH}) set(CUDA_STATIC_MODULES ${CUDA_STATIC_MODULES} ${alias} PARENT_SCOPE) + if (NOT ABS_PATH) + message(FATAL_ERROR "Can not find CUDA static library: ${file_name}") + endif() endfunction() add_cuda_static_lib(cudart_static CUDNN_CHECK_LIBRARY_DIRS libcudart_static.a) add_cuda_static_lib(cublas_static CUDNN_CHECK_LIBRARY_DIRS libcublas_static.a) add_cuda_static_lib(curand_static CUDNN_CHECK_LIBRARY_DIRS libcurand_static.a) add_cuda_static_lib(culibos_static CUDNN_CHECK_LIBRARY_DIRS libculibos.a) -if((${CUDA_VERSION} GREATER 10.0) OR (${CUDA_VERSION} EQUAL 10.0)) +if(NOT ${CUDA_VERSION} LESS 10.1) add_cuda_static_lib(cublasLt_static CUDNN_CHECK_LIBRARY_DIRS libcublasLt_static.a) endif()