|
|
@@ -94,7 +94,11 @@ if (CUDAToolkit_FOUND)
|
|
|
# As of 12.3.1 CUDA Toolkit for Windows does not offer a static cublas library
|
|
|
target_link_libraries(ggml-cuda PRIVATE CUDA::cudart_static CUDA::cublas)
|
|
|
else ()
|
|
|
- target_link_libraries(ggml-cuda PRIVATE CUDA::cudart_static CUDA::cublas_static)
|
|
|
+ if (CUDAToolkit_VERSION VERSION_GREATER_EQUAL "10.1")
|
|
|
+ target_link_libraries(ggml-cuda PRIVATE CUDA::cudart_static CUDA::cublas_static CUDA::cublasLt_static)
|
|
|
+ else()
|
|
|
+ target_link_libraries(ggml-cuda PRIVATE CUDA::cudart_static CUDA::cublas_static)
|
|
|
+ endif()
|
|
|
endif()
|
|
|
else()
|
|
|
target_link_libraries(ggml-cuda PRIVATE CUDA::cudart CUDA::cublas)
|