forked from unum-cloud/usearch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
build.rs
77 lines (69 loc) · 2.74 KB
/
build.rs
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
fn main() {
let mut build = cxx_build::bridge("rust/lib.rs");
build
.file("rust/lib.cpp")
.flag_if_supported("-Wno-unknown-pragmas")
.warnings(false)
.include("include")
.include("rust")
.include("fp16/include")
.include("simsimd/include");
build
.define("USEARCH_USE_SIMSIMD", "1")
.define("SIMSIMD_DYNAMIC_DISPATCH", "1")
.define("USEARCH_USE_OPENMP", "0")
.define("USEARCH_USE_FP16LIB", "0");
// Conditional compilation depending on the target operating system.
if cfg!(target_os = "linux") {
build
.flag_if_supported("-std=c++17")
.flag_if_supported("-O3")
.flag_if_supported("-ffast-math")
.flag_if_supported("-fdiagnostics-color=always")
.flag_if_supported("-g1"); // Simplify debugging
} else if cfg!(target_os = "macos") {
build
.flag_if_supported("-mmacosx-version-min=10.15")
.flag_if_supported("-std=c++17")
.flag_if_supported("-O3")
.flag_if_supported("-ffast-math")
.flag_if_supported("-fcolor-diagnostics")
.flag_if_supported("-g1"); // Simplify debugging
} else if cfg!(target_os = "windows") {
build
.flag_if_supported("/std:c++17")
.flag_if_supported("/O2")
.flag_if_supported("/fp:fast")
.flag_if_supported("/W1"); // Reduce warnings verbosity
}
if build.try_compile("usearch").is_err() {
print!("cargo:warning=Failed to compile with all SIMD backends...");
let target_arch = std::env::var("CARGO_CFG_TARGET_ARCH").unwrap_or_default();
let flags_to_try = match target_arch.as_str() {
"arm" | "aarch64" => vec!["SIMSIMD_TARGET_NEON", "SIMSIMD_TARGET_SVE"],
_ => vec![
"SIMSIMD_TARGET_SAPPHIRE",
"SIMSIMD_TARGET_ICE",
"SIMSIMD_TARGET_SKYLAKE",
"SIMSIMD_TARGET_HASWELL",
],
};
for flag in flags_to_try.iter() {
build.define(flag, "0");
if build.try_compile("usearch").is_ok() {
break;
}
// Print the failed configuration
println!(
"cargo:warning=Failed to compile after disabling {}, trying next configuration...",
flag
);
}
}
println!("cargo:rerun-if-changed=rust/lib.rs");
println!("cargo:rerun-if-changed=rust/lib.cpp");
println!("cargo:rerun-if-changed=rust/lib.hpp");
println!("cargo:rerun-if-changed=include/index_plugins.hpp");
println!("cargo:rerun-if-changed=include/index_dense.hpp");
println!("cargo:rerun-if-changed=include/usearch/index.hpp");
}