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

fix OpenCL build

Browse source
This commit is contained in:
Zero 2023-05-05 22:38:30 -06:00
parent 997bc936e4
commit e2f39c9c25
No known key found for this signature in database
GPG key ID: 3861E636EA1E0E2B
1 changed files with 8 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

11
sys/build.rs
View file

@ -26,6 +26,11 @@ fn main() {
println!("cargo:rustc-link-lib=static=whisper");
#[cfg(feature = "coreml")]
println!("cargo:rustc-link-lib=static=whisper.coreml");
#[cfg(feature = "opencl")]
{
println!("cargo:rustc-link-lib=clblast");
println!("cargo:rustc-link-lib=OpenCL");
}
println!("cargo:rerun-if-changed=wrapper.h");
if env::var("WHISPER_DONT_GENERATE_BINDINGS").is_ok() {
@ -81,13 +86,13 @@ fn main() {
.arg("-DWHISPER_BUILD_EXAMPLES=OFF");
#[cfg(feature = "coreml")]
cmd.arg("-DWHISPER_COREML=1");
cmd.arg("-DWHISPER_COREML=ON");
#[cfg(feature = "cuda")]
cmd.arg("-DWHISPER_CUBLAS=1");
cmd.arg("-DWHISPER_CUBLAS=ON");
#[cfg(feature = "opencl")]
cmd.arg("-DWHISPER_CLBLAST=1");
cmd.arg("-DWHISPER_CLBLAST=ON");
let code = cmd.status().expect("Failed to run `cmake`");
if code.code() != Some(0) {