1 files changed, 20 insertions, 0 deletions
diff --git a/ext/hip_runtime-sys/build.rs b/ext/hip_runtime-sys/build.rs
new file mode 100644
index 0000000..c63b526
--- /dev/null
+++ b/ext/hip_runtime-sys/build.rs
@@ -0,0 +1,20 @@
+use std::env::VarError;
+use std::{env, path::PathBuf};
+
+fn main() -> Result<(), VarError> {
+    if cfg!(windows) {
+        println!("cargo:rustc-link-lib=dylib=amdhip64_6");
+        let env = env::var("CARGO_CFG_TARGET_ENV")?;
+        if env == "msvc" {
+            let mut path = PathBuf::from(env::var("CARGO_MANIFEST_DIR")?);
+            path.push("lib");
+            println!("cargo:rustc-link-search=native={}", path.display());
+        } else {
+            println!("cargo:rustc-link-search=native=C:\\Windows\\System32");
+        };
+    } else {
+        println!("cargo:rustc-link-lib=dylib=amdhip64");
+        println!("cargo:rustc-link-search=native=/opt/rocm/lib/");
+    }
+    Ok(())
+}