fix: link openblas path correctly

tazz4843 · Aug 1, 2024 · 67924ca · 67924ca
1 parent 744804a
commit 67924ca
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/sys/build.rs b/sys/build.rs
@@ -37,7 +37,17 @@ fn main() {
     }
     #[cfg(feature = "openblas")]
     {
-        println!("cargo:rustc-link-lib=openblas");
+        if let Ok(openblas_path) = env::var("OPENBLAS_PATH") {
+            println!(
+                "cargo::rustc-link-search={}",
+                PathBuf::from(openblas_path).join("lib").display()
+            );
+        }
+        if cfg!(windows) {
+            println!("cargo:rustc-link-lib=libopenblas");
+        } else {
+            println!("cargo:rustc-link-lib=openblas");
+        }
     }
     #[cfg(feature = "cuda")]
     {