From 9309e0314f00283ca5f4dd0115b347c6adc8f10a Mon Sep 17 00:00:00 2001 From: Meng Zhang Date: Fri, 27 Oct 2023 21:25:45 -0700 Subject: [PATCH] fix: fix docker build --- crates/llama-cpp-bindings/build.rs | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/crates/llama-cpp-bindings/build.rs b/crates/llama-cpp-bindings/build.rs index 59a2106..3c924ee 100644 --- a/crates/llama-cpp-bindings/build.rs +++ b/crates/llama-cpp-bindings/build.rs @@ -3,6 +3,8 @@ use cmake::Config; fn main() { println!("cargo:rerun-if-changed=cc/*.h"); println!("cargo:rerun-if-changed=cc/*.cc"); + println!("cargo:rustc-link-lib=llama"); + println!("cargo:rustc-link-lib=ggml_static"); let mut config = Config::new("llama.cpp"); if cfg!(target_os = "macos") { @@ -15,7 +17,6 @@ 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"); @@ -24,8 +25,6 @@ fn main() { let dst = config.build(); println!("cargo:rustc-link-search=native={}/build", dst.display()); - println!("cargo:rustc-link-lib=llama"); - println!("cargo:rustc-link-lib=ggml_static"); cxx_build::bridge("src/lib.rs") .file("src/engine.cc")