build.rs

use juno_build::JunoCompiler;

fn main() {
    JunoCompiler::new()
        .ir_in_src("matmul.hir")
        .unwrap()
        .build()
        .unwrap();

    #[cfg(feature = "cuda")]
    println!("cargo::rustc-link-search=native=/usr/lib/x86_64-linux-gnu/");
    #[cfg(feature = "cuda")]
    println!("cargo:rustc-link-search=native=/usr/local/cuda/lib64");
    #[cfg(feature = "cuda")]
    println!("cargo:rustc-link-lib=cudart");
}