ci: fix llama-server build

TabbyML · Jul 3, 2024 · 34799a5 · 34799a5
1 parent 7b048c2
commit 34799a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crates/llama-cpp-server/build.rs b/crates/llama-cpp-server/build.rs
@@ -22,7 +22,7 @@ fn main() {
         println!("cargo:rustc-link-lib=framework=MetalKit");
     }
     if cfg!(feature = "cuda") {
-        config.define("LLAMA_CUBLAS", "ON");
+        config.define("GGML_CUDA", "ON");
         config.define("CMAKE_POSITION_INDEPENDENT_CODE", "ON");
     }
     if cfg!(feature = "rocm") {