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

feat: add vulkan with new whisper.cpp structure

Browse source 
Co-authored-by: newfla <newfla@hotmail.it>
This commit is contained in:
thewh1teagle 2024-08-18 21:23:46 +03:00
parent 044e085555
commit e33cc3eb22
No known key found for this signature in database
5 changed files with 22 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

18
sys/build.rs
View file

@ -166,6 +166,24 @@ fn main() {
}
}
if cfg!(feature = "vulkan") {
config.define("GGML_VULKAN", "ON");
if cfg!(windows) {
println!("cargo:rerun-if-env-changed=VULKAN_SDK");
println!("cargo:rustc-link-lib=vulkan-1");
let vulkan_path = match env::var("VULKAN_SDK") {
Ok(path) => PathBuf::from(path),
Err(_) => panic!(
"Please install Vulkan SDK and ensure that VULKAN_SDK env variable is set"
),
};
let vulkan_lib_path = vulkan_path.join("Lib");
println!("cargo:rustc-link-search={}", vulkan_lib_path.display());
} else {
println!("cargo:rustc-link-lib=vulkan");
}
}
if cfg!(feature = "openblas") {
config.define("GGML_BLAS", "ON");
}