zksync-gpu-ffi 0.140.0-gpu-wrapper.0

ZKsync GPU FFI
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

extern crate bindgen;
use std::{env, path::PathBuf};

// build.rs

fn main() {
    let bellman_cuda_path = if let Ok(path) = std::env::var("BELLMAN_CUDA_DIR") {
        path
    } else {
        // we need to instruct rustc so that it will find libbellman-cuda.a
        //   - if dep is resolved via git(cargo checks ~/.cargo/git/checkouts/)
        //   - if dep is resolved via local path
        //   - if you want to build on a macos or only for rust analyzer
        //      just `export BELLMAN_CUDA_DIR=$PWD/bellman-cuda`
        // so we will benefit from env variable for now
        todo!("set BELLMAN_CUDA_DIR=$PWD")
    };

    generate_bindings(&bellman_cuda_path);

    #[cfg(not(target_os = "macos"))]
    link_multiexp_library(&bellman_cuda_path); // FIXME enable
}

fn generate_bindings(bellman_cuda_path: &str) {
    println!("generating bindings");
    let header_file = &format!("{}/src/bellman-cuda.h", bellman_cuda_path);
    const OUT_FILE: &str = "bindings.rs";
    println!("cargo:rerun-if-changed={}", header_file);

    let bindings = bindgen::Builder::default()
        // The input header we would like to generate
        // bindings for.
        .header(header_file)
        // Tell cargo to invalidate the built crate whenever any of the
        // included header files changed.
        .parse_callbacks(Box::new(bindgen::CargoCallbacks))
        // Finish the builder and generate the bindings.
        .generate()
        // Unwrap the Result and panic on failure.
        .expect("Unable to generate bindings");

    // Write the bindings to the $OUT_DIR/bindings.rs file.
    let out_path = PathBuf::from(format!(
        "{}/{}",
        env::current_dir().unwrap().to_str().unwrap(),
        "src"
    ));
    println!("out path {:?}", out_path.to_str());
    bindings
        .write_to_file(out_path.join(OUT_FILE))
        .expect("Couldn't write bindings!");
}

fn link_multiexp_library(bellman_cuda_path: &str) {
    let kind = "static";
    let name = "bellman-cuda";

    println!("cargo:rustc-link-lib=dylib=stdc++");
    println!("cargo:rustc-link-search=native=/usr/local/cuda/lib64");
    println!("cargo:rustc-link-lib=dylib=cudart");
    println!("cargo:rustc-link-lib=static=cudadevrt");
    println!(
        "cargo:rustc-link-search=native={}/build/src",
        bellman_cuda_path
    );
    println!("cargo:rustc-link-lib={}={}", kind, name);
}