harald/wyoming-whisper-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

link cuda when cuda feature is on

Browse source
This commit is contained in:
hyiip 2025-02-19 02:55:46 +08:00
parent 3c20758cce
commit 3d33a73458
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
sys/build.rs
View file

@ -265,6 +265,10 @@ fn main() {
if cfg!(feature = "metal") { if cfg!(feature = "metal") {
println!("cargo:rustc-link-lib=static=ggml-metal"); println!("cargo:rustc-link-lib=static=ggml-metal");
} }
if cfg!(feature = "cuda") {
println!("cargo:rustc-link-lib=static=ggml-cuda");
}
println!( println!(
"cargo:WHISPER_CPP_VERSION={}", "cargo:WHISPER_CPP_VERSION={}",