fn main() {
if cfg!(windows) {
let known_sdk = [
("OPENCL_PATH", "", ""),
("OPENCL_ROOT", "x64", "x86"),
("INTELOCLSDKROOT", "x64", "x86"),
("AMDAPPSDKROOT", "x86_64", "x86"),
("CUDA_PATH", "x64", "Win32"),
];
for info in &known_sdk {
if let Ok(sdk) = std::env::var(info.0) {
let mut path = std::path::PathBuf::from(sdk);
path.push("lib");
if cfg!(target_arch = "x86_64") {
if !info.1.is_empty() {
path.push(info.1);
}
} else if !info.2.is_empty() {
path.push(info.2);
}
println!("cargo:rustc-link-search=native={}", path.display());
break;
}
}
}
}