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

Add a feature flag for Metal acceleration support

Browse source
This commit is contained in:
Niko 2023-10-09 17:02:54 -06:00
parent 7dac364460
commit 253ac1bb6f
No known key found for this signature in database
GPG key ID: 3861E636EA1E0E2B
3 changed files with 17 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

15
sys/build.rs
View file

@ -20,6 +20,12 @@ fn main() {
println!("cargo:rustc-link-lib=framework=Foundation");
println!("cargo:rustc-link-lib=framework=CoreML");
}
#[cfg(feature = "metal")]
{
println!("cargo:rustc-link-lib=framework=Foundation");
println!("cargo:rustc-link-lib=framework=Metal");
println!("cargo:rustc-link-lib=framework=MetalKit");
}
}
println!("cargo:rustc-link-search={}", env::var("OUT_DIR").unwrap());
@ -118,6 +124,15 @@ fn main() {
#[cfg(feature = "opencl")]
cmd.arg("-DWHISPER_CLBLAST=ON");
cfg_if! {
if #[cfg(feature = "metal")] {
cmd.arg("-DWHISPER_METAL=ON");
} else {
// Metal is enabled by default so we need to explicitly disable it
cmd.arg("-DWHISPER_METAL=OFF");
}
};
cmd.arg("-DCMAKE_POSITION_INDEPENDENT_CODE=ON");