russell_lab 0.7.2

Matrix-vector laboratory including linear algebra tools
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

use std::env;

const MKL_VERSION: &str = "2023.2.0";

fn main() {
    // math functions
    cc::Build::new().file("c_code/math_functions.c").compile("c_code");

    // option
    let use_intel_mkl = match env::var("RUSSELL_LAB_USE_INTEL_MKL") {
        Ok(v) => v == "1" || v.to_lowercase() == "true",
        Err(_) => false,
    };

    if use_intel_mkl {
        // Intel MKL
        cc::Build::new()
            .file("c_code/interface_blas.c")
            .include(format!("/opt/intel/oneapi/mkl/{}/include", MKL_VERSION))
            .define("USE_INTEL_MKL", None)
            .compile("c_code_interface_blas");
        println!(
            "cargo:rustc-link-search=native=/opt/intel/oneapi/mkl/{}/lib/intel64",
            MKL_VERSION
        );
        println!(
            "cargo:rustc-link-search=native=/opt/intel/oneapi/compiler/{}/linux/compiler/lib/intel64_lin",
            MKL_VERSION
        );
        println!("cargo:rustc-link-lib=mkl_intel_lp64");
        println!("cargo:rustc-link-lib=mkl_intel_thread");
        println!("cargo:rustc-link-lib=mkl_core");
        println!("cargo:rustc-link-lib=pthread");
        println!("cargo:rustc-link-lib=m");
        println!("cargo:rustc-link-lib=dl");
        println!("cargo:rustc-link-lib=iomp5");
        println!("cargo:rustc-cfg=use_intel_mkl");
    } else {
        // OpenBLAS
        cc::Build::new()
            .file("c_code/interface_blas.c")
            .compile("c_code_interface_blas");
        println!("cargo:rustc-link-lib=dylib=openblas");
        println!("cargo:rustc-link-lib=dylib=lapack");
    }
}