mathic 0.2.0

A compiler with builtin support of symbolic operations, built with LLVM/MLIR
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

use std::env;
use std::path::PathBuf;
use std::process::Command;

fn main() {
    let root = PathBuf::from(env::var("CARGO_MANIFEST_DIR").unwrap());
    let dialects_root = root.join("Dialects");
    let build_dir = PathBuf::from(env::var("OUT_DIR").unwrap()).join("DialectBuild/");
    let llvm_prefix = env::var("LLVM_SYS_211_PREFIX").unwrap();

    std::fs::create_dir_all(&build_dir).unwrap();

    // ── Build ───────────────────────────────────────────────────────────────

    let status = Command::new("cmake")
        .arg(&dialects_root)
        .arg(format!("-DMLIR_DIR={}/lib/cmake/mlir", llvm_prefix))
        .arg(format!("-DLLVM_DIR={}/lib/cmake/llvm", llvm_prefix))
        .arg("-S")
        .arg(&dialects_root)
        .arg("-B")
        .arg(&build_dir)
        .arg("-DCMAKE_POLICY_VERSION_MINIMUM=3.10")
        .status()
        .expect("cmake configure failed");
    assert!(status.success(), "cmake configure failed");

    let status = Command::new("cmake")
        .arg("--build")
        .arg(&build_dir)
        .status()
        .expect("cmake build failed");
    assert!(status.success(), "cmake build failed");

    // ── Link ────────────────────────────────────────────────────────────────

    let lib_dir = build_dir.join("lib/");

    // Make the linker aware of the dialect's library dir.
    println!("cargo:rustc-link-arg=-Wl,-rpath,{}", lib_dir.display());
    println!("cargo:rustc-link-arg=-Wl,-rpath,{}/lib", llvm_prefix);

    #[cfg(target_arch = "aarch64")]
    println!("cargo:rustc-link-arg=-Wl,-rpath,/opt/homebrew/lib");

    println!("cargo:rustc-link-search=native={}", lib_dir.display());
    println!("cargo:rustc-link-lib=dylib=MLIRSymbolicDialect");
    println!("cargo:rustc-link-lib=dylib=MLIRSymbolicTransforms");
    println!("cargo:rustc-link-lib=dylib=MLIRSymbolicToArith");
    println!("cargo:rustc-link-lib=dylib=dialect_bindings");

    // ── Rerun triggers ──────────────────────────────────────────────────────

    println!("cargo:rerun-if-changed=CMakeLists.txt");
    println!("cargo:rerun-if-changed=Dialects/include");
    println!("cargo:rerun-if-changed=Dialects/lib");
    println!("cargo:rerun-if-changed=build.rs");
}