fix: fix cuda undefined reference error by adding cuda linkage (#1286)

TabbyML · Jan 24, 2024 · 22e0175 · 22e0175
1 parent c32398a
commit 22e0175
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/crates/llama-cpp-bindings/build.rs b/crates/llama-cpp-bindings/build.rs
@@ -40,6 +40,7 @@ fn build_llama_cpp() {
             println!("cargo:rustc-link-search=native=/usr/local/cuda/lib64");
             println!("cargo:rustc-link-lib=culibos");
         }
+        println!("cargo:rustc-link-lib=cuda");
         println!("cargo:rustc-link-lib=cudart");
         println!("cargo:rustc-link-lib=cublas");
         println!("cargo:rustc-link-lib=cublasLt");