From 6dd12ce1ec361fba71b05db2f17adb434faa9cfc Mon Sep 17 00:00:00 2001
From: Meng Zhang <meng@tabbyml.com>
Date: Fri, 27 Oct 2023 19:40:35 -0700
Subject: [PATCH] fix: adding cuda search path to docker build.

---
 crates/llama-cpp-bindings/build.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/crates/llama-cpp-bindings/build.rs b/crates/llama-cpp-bindings/build.rs
index a0c511b..59a2106 100644
--- a/crates/llama-cpp-bindings/build.rs
+++ b/crates/llama-cpp-bindings/build.rs
@@ -14,7 +14,10 @@ fn main() {
     }
     if cfg!(feature = "cuda") {
         config.define("LLAMA_CUBLAS", "ON");
+        println!("cargo:rustc-link-search=native=/usr/local/cuda/lib64");
+        println!("cargo:rustc-link-search=native=/opt/cuda/lib64");
         println!("cargo:rustc-link-lib=cudart");
+        println!("cargo:rustc-link-lib=culibos");
         println!("cargo:rustc-link-lib=cublas");
         println!("cargo:rustc-link-lib=cublasLt");
     }