]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
Link to cublas dynamically on Windows even with LLAMA_STATIC (#4506)
authorBach Le <redacted>
Sun, 17 Dec 2023 10:57:33 +0000 (18:57 +0800)
committerGitHub <redacted>
Sun, 17 Dec 2023 10:57:33 +0000 (11:57 +0100)
CMakeLists.txt

index 57b43c136fd5e34e607822dbcdc91de6a7e67d2c..e3cd43ab36f0626efbfd74335d5e65a8f81dcaf8 100644 (file)
@@ -291,7 +291,12 @@ if (LLAMA_CUBLAS)
         add_compile_definitions(GGML_CUDA_PEER_MAX_BATCH_SIZE=${LLAMA_CUDA_PEER_MAX_BATCH_SIZE})
 
         if (LLAMA_STATIC)
-            set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} CUDA::cudart_static CUDA::cublas_static CUDA::cublasLt_static)
+            if (WIN32)
+                # As of 12.3.1 CUDA Tookit for Windows does not offer a static cublas library
+                set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} CUDA::cudart_static CUDA::cublas CUDA::cublasLt)
+            else ()
+                set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} CUDA::cudart_static CUDA::cublas_static CUDA::cublasLt_static)
+            endif()
         else()
             set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} CUDA::cudart CUDA::cublas CUDA::cublasLt)
         endif()