kannolo 0.3.4

kANNolo is designed for easy prototyping of ANN Search algorithms while ensuring high effectiveness and efficiency over both dense and sparse vectors.
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::fs;
use std::path::{Path, PathBuf};

fn main() {
    println!("cargo::rustc-check-cfg=cfg(use_cblas)");

    if cblas_is_available() {
        println!("cargo:rustc-cfg=use_cblas");
        println!("Detected MKL/CBLAS");
    } else {
        println!("Did not detect MKL/CBLAS");
    }
}

fn cblas_is_available() -> bool {
    if let Ok(mklroot) = std::env::var("MKLROOT") {
        if check_mkl_directory(&mklroot) {
            return true;
        }
    } else {
        println!("MKLROOT is not set");
        // MKLROOT is not set
        return false;
    }

    println!("Checking for MKL/CBLAS");
    let base_path = PathBuf::from("/opt/intel/oneapi/mkl/");
    if base_path.exists() {
        if let Ok(entries) = fs::read_dir(base_path) {
            for entry in entries.filter_map(|e| e.ok()) {
                let path = entry.path();
                if path.is_dir() && check_mkl_directory(&path.to_string_lossy()) {
                    println!("using blas!");
                    return true;
                }
            }
        }
    }

    false
}

fn check_mkl_directory(mklroot: &str) -> bool {
    let lib_path = PathBuf::from(mklroot).join("lib/intel64/libmkl_rt.so");
    Path::new(&lib_path).exists()
}