cubecl-hip-sys 7.0.5183101

Rust bindings for AMD ROCm HIP runtime libraries used by CubeCL.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

include!("src/hipconfig.rs");

const HIP_FEATURE_PREFIX: &str = "CARGO_FEATURE_HIP_";

/// Make sure that at least one and only one hip feature is set.
/// If None are set then we use the passed defautl version to set the corresponding feature.
/// Returns the selected HIP patch version.
fn set_hip_feature(default_version: &str) {
    let mut enabled_features = Vec::new();

    for (key, value) in std::env::vars() {
        if key.starts_with(HIP_FEATURE_PREFIX) && value == "1" {
            enabled_features.push(format!(
                "hip_{}",
                key.strip_prefix(HIP_FEATURE_PREFIX).unwrap()
            ));
        }
    }

    if enabled_features.is_empty() {
        let default_hip_feature = format!("hip_{default_version}");
        println!("cargo:rustc-cfg=feature=\"{default_hip_feature}\"");
    } else {
        panic!("Error: HIP_XXX feature detected!\nHIP_XXX features should not be set manually. Remove the feature and change your HIP_PATH environment variable instead.");
    }
}

fn main() {
    println!("cargo::rerun-if-changed=build.rs");
    println!("cargo::rerun-if-env-changed=ROCM_PATH");
    println!("cargo::rerun-if-env-changed=HIP_PATH");
    let hip_system_patch = get_hip_patch_version();
    if let Ok(ref patch) = hip_system_patch {
        set_hip_feature(patch);
        println!("cargo::rustc-link-lib=dylib=hiprtc");
        println!("cargo::rustc-link-lib=dylib=amdhip64");
        let lib_path = get_hip_ld_library_path().unwrap();
        println!("cargo::rustc-link-search=native={lib_path}");
    } else {
        // There is no 'hipconfig' on the system, so we assume there is no HIP installation available on the system.
        // Nevertheless we still want crates that depend on the bindings to compile even if they don't need to
        // link against the HIP libraries, especially for cargo clippy.
        // We decide to set the last version of HIP bindings as the default for this purpose, i.e. the HIP version that
        // corresponds to last published version of 'cubecl-hip-sys'.
        let feature = extract_latest_hip_feature_from_path("Cargo.toml").unwrap();
        println!(
            "cargo::warning=Defaulting to the latest feature of HIP bindings available: {feature}"
        );
        println!("cargo:rustc-cfg=feature=\"{feature}\"");
    }
}