surrealml-core 0.0.4

The core machine learning library for SurrealML that enables SurrealDB to store and load ML models
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

use std::path::Path;
use std::env;
use std::fs;


fn main() -> std::io::Result<()> {

    match std::env::var("ONNXRUNTIME_LIB_PATH") {
        Ok(_) => {
            println!("cargo:rustc-cfg=onnx_runtime_env_var_set");
        },
        Err(_) => {
            let target_lib = match env::var("CARGO_CFG_TARGET_OS").unwrap() {
                ref s if s.contains("linux") => "libonnxruntime.so",
                ref s if s.contains("macos") => "libonnxruntime.dylib",
                ref s if s.contains("windows") => "onnxruntime.dll",
                // ref s if s.contains("android") => "android", => not building for android
                _ => panic!("Unsupported target os")
            };
            let profile = match env::var("PROFILE").unwrap() {
                ref s if s.contains("release") => "release",
                ref s if s.contains("debug") => "debug",
                _ => panic!("Unsupported profile")
            };
            let lib_path = Path::new("target").join(profile).join(target_lib);
            // put it next to the file of the embedding
            let destination = Path::new(target_lib);
            fs::copy(lib_path, destination)?;
        }

    }
    Ok(())
}