root
/
tabby
mirror of https://gitee.com/zhang_1334717033/tabby.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: docker build for llama cuda backend

release-notes-05
Meng Zhang 2023-10-27 16:36:54 -07:00 committed by GitHub
parent c3e7b470e5
commit 2d948639be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
crates/llama-cpp-bindings/build.rs
View File

@ -14,6 +14,9 @@ fn main() {
} }
if cfg!(feature = "cuda") { if cfg!(feature = "cuda") {
config.define("LLAMA_CUBLAS", "ON"); config.define("LLAMA_CUBLAS", "ON");
println!("cargo:rustc-link-lib=cudart");
println!("cargo:rustc-link-lib=cublas");
println!("cargo:rustc-link-lib=cublasLt");
} }
let dst = config.build(); let dst = config.build();