15 lines
423 B
Rust
15 lines
423 B
Rust
|
use std::env::VarError;
|
||
|
|
use std::{env, path::PathBuf};
|
||
|
|
|
||
|
|
fn main() -> Result<(), VarError> {
|
||
|
|
println!("cargo:rustc-link-lib=dylib=hiprtc");
|
||
|
|
if cfg!(windows) {
|
||
|
|
let mut path = PathBuf::from(env::var("HIP_PATH")?);
|
||
|
|
path.push("lib");
|
||
|
|
println!("cargo:rustc-link-search=native={}", path.display());
|
||
|
|
} else {
|
||
|
|
println!("cargo:rustc-link-search=native=/opt/rocm/lib/");
|
||
|
|
}
|
||
|
|
Ok(())
|
||
|
|
}
|