1 files changed, 19 insertions, 0 deletions
diff --git a/hip_runtime-sys/build.rs b/hip_runtime-sys/build.rs
new file mode 100644
index 0000000..b6d842e
--- /dev/null
+++ b/hip_runtime-sys/build.rs
@@ -0,0 +1,19 @@
+use std::env::VarError;
+use std::{env, path::PathBuf};
+
+fn main() -> Result<(), VarError> {
+    println!("cargo:rustc-link-lib=dylib=amdhip64");
+    if cfg!(windows) {
+        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-search=native=/opt/rocm/lib/");
+    }
+    Ok(())
+}