fix: add missing search path for link opencl libs

tazz4843 · Aug 1, 2024 · d1eea3a · d1eea3a
1 parent 744804a
commit d1eea3a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -1,4 +1,7 @@
 **/target
 **/Cargo.lock
 /.idea
-/.vscode
+/.vscode
+/clblast
+*.7z
+*.bin
diff --git a/sys/build.rs b/sys/build.rs
@@ -32,6 +32,12 @@ fn main() {
     println!("cargo:rustc-link-lib=static=whisper.coreml");
     #[cfg(feature = "opencl")]
     {
+        if let Ok(clblast_dir) = env::var("CLBLast_DIR") {
+            println!("cargo::rustc-link-search={}", PathBuf::from(clblast_dir).join("..\\..").display());
+        }
+        if let Ok(opencl_dir) = env::var("OPENCL_DIR") {
+            println!("cargo::rustc-link-search={}", opencl_dir);
+        }
         println!("cargo:rustc-link-lib=clblast");
         println!("cargo:rustc-link-lib=OpenCL");
     }