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

Merge pull request #170 from thewh1teagle/feat/vulkan

Browse source 
Feat/vulkan
This commit is contained in:
Niko 2024-09-13 21:07:32 +00:00 • committed by GitHub
commit d8b6aefe06
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 21 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

18
sys/build.rs
View file

@ -178,6 +178,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");
}