autograph 0.0.2

Machine Learning Library for Rust
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

#![allow(warnings)]
use std::{env, ffi::CString, process::Command};

fn main() {
    {
        // dnnl
        let dst = cmake::Config::new("oneDNN")
            .define("DNNL_LIBRARY_TYPE", "STATIC")
            .define("DNNL_BUILD_EXAMPLES", "OFF")
            .define("DNNL_BUILD_TESTS", "OFF")
            .build();
        println!(
            "cargo:rustc-link-search=native={}",
            dst.join("lib").display()
        );
        println!("cargo:rustc-link-lib=static=dnnl");
        if cfg!(target_os = "linux") {
            println!("cargo:rustc-link-lib=dylib=gomp");
        }
        cpp_build::Config::new()
            .include(dst.join("include").display().to_string())
            .build("src/lib.rs");
    }
    #[cfg(feature = "cuda")]
    {
        {
            // compile custom cuda source
            println!("cargo:rustc-rerun-if-changed=src/cuda/kernels.cu");
            let status = Command::new("nvcc")
                .arg("src/cuda/kernels.cu")
                .arg("--ptx")
                .arg("-odir")
                .arg(env::var("OUT_DIR").unwrap())
                .status()
                .unwrap();
            assert!(status.success());
        }
    }
}