From e2f39c9c257ee93a2b02cc3ac94f4e22a871cd4c Mon Sep 17 00:00:00 2001
From: Zero <zero@imaskeleton.me>
Date: Fri, 5 May 2023 22:38:30 -0600
Subject: [PATCH] fix OpenCL build

---
 sys/build.rs | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/sys/build.rs b/sys/build.rs
index b5f2e89..9d3d658 100644
--- a/sys/build.rs
+++ b/sys/build.rs
@@ -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) {