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

cfg_if

Browse source
This commit is contained in:
jnnnnn 2023-05-29 19:17:48 +12:00
parent ffa9655a06
commit a4e303ef3f
No known key found for this signature in database
1 changed files with 9 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

9
sys/build.rs
View file

@ -36,18 +36,17 @@ fn main() {
println!("cargo:rustc-link-lib=cublas"); println!("cargo:rustc-link-lib=cublas");
println!("cargo:rustc-link-lib=cudart"); println!("cargo:rustc-link-lib=cudart");
println!("cargo:rustc-link-lib=cublasLt"); println!("cargo:rustc-link-lib=cublasLt");
#[cfg(target_os = "windows")] cfg_if! {
{ if #[cfg(target_os = "windows")] {
let cuda_path = PathBuf::from(env::var("CUDA_PATH").unwrap()).join("lib/x64"); let cuda_path = PathBuf::from(env::var("CUDA_PATH").unwrap()).join("lib/x64");
println!("cargo:rustc-link-search={}", cuda_path.display()); println!("cargo:rustc-link-search={}", cuda_path.display());
} } else {
#[cfg(not(target_os = "windows"))]
{
println!("cargo:rustc-link-lib=culibos"); println!("cargo:rustc-link-lib=culibos");
println!("cargo:rustc-link-search=/usr/local/cuda/lib64"); println!("cargo:rustc-link-search=/usr/local/cuda/lib64");
println!("cargo:rustc-link-search=/opt/cuda/lib64"); println!("cargo:rustc-link-search=/opt/cuda/lib64");
} }
} }
}
println!("cargo:rerun-if-changed=wrapper.h"); println!("cargo:rerun-if-changed=wrapper.h");
if env::var("WHISPER_DONT_GENERATE_BINDINGS").is_ok() { if env::var("WHISPER_DONT_GENERATE_BINDINGS").is_ok() {