onednn-sys 0.0.1

Rust bindings for oneDNN
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

use std::{error::Error, fs::File, io::{BufRead, BufReader}, path::Path};

fn main() -> Result<(), Box<dyn Error>> {
    let dst = cmake::Config::new("oneDNN")
        .define("DNNL_LIBRARY_TYPE", "STATIC")
        .define("DNNL_BUILD_EXAMPLES", "OFF")
        .define("DNNL_BUILD_TESTS", "OFF")
        .define("DNNL_CPU_RUNTIME", "OMP")
        .define("DNNL_GPU_RUNTIME", "NONE")
        .build();
    println!(
        "cargo:rustc-link-search=native={}",
        dst.join("lib").display()
    );
    println!(
        "cargo:rustc-link-search=native={}",
        dst.join("lib64").display()
    );
    println!("cargo:rustc-link-lib=static=dnnl");
    let mut omp_library = None;
    let mut omp_lib_name = None;
    
    let cmake_cache_path = dst.join("build").join("CMakeCache.txt");
    let cmake_cache_reader = BufReader::new(File::open(cmake_cache_path)?);
  
    let omp_lib_name_pfx = "OpenMP_CXX_LIB_NAMES:STRING=";
    let mut omp_library_pfx = None;
    
    let mut cmake_cache = String::new();
    
    for line in cmake_cache_reader.lines()
        .filter_map(|line| line.ok()) { 
        cmake_cache.push_str(&format!("{}\n", line));
            
        if line.starts_with(omp_lib_name_pfx) {
            let _omp_lib_name = line[omp_lib_name_pfx.len()..]
                .split(";")
                .take(1)
                .next();
            if let Some(_omp_lib_name) = _omp_lib_name {
                omp_lib_name.replace(String::from(_omp_lib_name));
                omp_library_pfx.replace(format!("OpenMP_{}_LIBRARY:FILEPATH=", _omp_lib_name));
            }
        }
        else if let Some(ref omp_library_pfx) = omp_library_pfx {
            if line.starts_with(&*omp_library_pfx) {
                omp_library.replace(String::from(&line[omp_library_pfx.len()..]));
            }
        }
    }
    
    if let (Some(omp_library), Some(omp_lib_name)) = (omp_library, omp_lib_name) {
        let omp_link_path = Path::new(&omp_library)
            .parent()
            .unwrap();
        let omp_link_path = omp_link_path.to_str().unwrap();
        println!("cargo:rustc-link-search={}", omp_link_path);
        println!("cargo:rustc-link-lib={}", omp_lib_name);
        println!("cargo:omp={}", omp_lib_name);
        println!("cargo:omp_lib={}", omp_link_path);
    }
    else {
        println!("cargo:warning=Unable to use OpenMP, running in SEQ mode. Performance on cpu will be signficantly reduced."); 
    }
    
    #[cfg(feature = "bindgen")]
    {
        let bindings = bindgen::Builder::default()
            .header("wrapper.hpp")
            .clang_arg("--std")
            .clang_arg("c++14")
            .clang_arg("-I")
            .clang_arg(dst.join("include").display().to_string())
            .ctypes_prefix("::libc")
            .generate_block(false)
            .size_t_is_usize(true)
            .rustified_non_exhaustive_enum("dnnl.*")
            .parse_callbacks(Box::new(bindgen::CargoCallbacks))
            .rustfmt_bindings(true)
            .generate()
            .expect("Unable to create bindings.");
        bindings.write_to_file("src/bindings.rs").unwrap();
    }
    
    if cfg!(target_os = "windows") {
        // nothing
    }
    else if cfg!(target_os = "macos") {
        println!("cargo:rustc-link-lib=c++");
    }
    else {
        println!("cargo:rustc-link-lib=stdc++");
    };
    
    
    
    println!("cargo:include={}", dst.join("include").display().to_string());
    
    println!("cargo:rustc-rerun-if-changed=wrapper.hpp");
    
    Ok(())
}