cndrv 0.1.2

Safe Cambricon driver API.
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
52
53
54
55
56

fn main() {
    use build_script_cfg::Cfg;
    use search_neuware_tools::find_neuware_home;
    use std::{env, path::PathBuf};

    println!("cargo:rerun-if-changed=build.rs");
    println!("cargo:rerun-if-env-changed=NEUWARE_HOME");

    let neuware = Cfg::new("detected_neuware");
    let Some(neuware_home) = find_neuware_home() else {
        return;
    };
    neuware.define();

    println!(
        "cargo:rustc-link-search=native={}",
        neuware_home.join("lib64").display()
    );
    println!("cargo:rustc-link-lib=dylib=cndrv");
    println!("cargo:rustc-link-lib=dylib=cnrtc");

    // Tell cargo to invalidate the built crate whenever the wrapper changes.
    println!("cargo:rerun-if-changed=wrapper.h");

    // The bindgen::Builder is the main entry point to bindgen,
    // and lets you build up options for the resulting bindings.
    let bindings = bindgen::Builder::default()
        // The input header we would like to generate bindings for.
        .header("wrapper.h")
        .clang_arg(format!("-I{}", neuware_home.join("include").display()))
        // Only generate bindings for the functions in these namespaces.
        .allowlist_function("cn.*")
        .allowlist_function("cnrtc.*")
        .allowlist_item("CN.*")
        // Annotate the given type with the #[must_use] attribute.
        .must_use_type("CNresult")
        .must_use_type("cnrtcStatus")
        // Generate rust style enums.
        .default_enum_style(bindgen::EnumVariation::Rust {
            non_exhaustive: true,
        })
        // Use core instead of std in the generated bindings.
        .use_core()
        // Tell cargo to invalidate the built crate whenever any of the included header files changed.
        .parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
        // Finish the builder and generate the bindings.
        .generate()
        // Unwrap the Result and panic on failure.
        .expect("Unable to generate bindings");

    // Write the bindings to the $OUT_DIR/bindings.rs file.
    let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
    bindings
        .write_to_file(out_path.join("bindings.rs"))
        .expect("Couldn't write bindings!");
}