forked from DelphinusLab/zkWasm-prover
-
Notifications
You must be signed in to change notification settings - Fork 0
/
build.rs
25 lines (20 loc) · 815 Bytes
/
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
fn main() {
extern crate cc;
cc::Build::new()
.cuda(true)
.flag("-cudart=shared")
.flag("-gencode")
.flag("arch=compute_89,code=sm_89")
.file("cuda/bn254.cu")
.compile("libzkwasm_prover_kernel.a");
/* Link CUDA Runtime (libcudart.so) */
// Add link directory
// - This path depends on where you install CUDA (i.e. depends on your Linux distribution)
// - This should be set by `$LIBRARY_PATH`
println!("cargo:rerun-if-changed={}", "cuda");
println!("cargo:rustc-link-search=native=/usr/local/cuda/lib64");
println!("cargo:rustc-link-lib=cudart");
/* Optional: Link CUDA Driver API (libcuda.so) */
// println!("cargo:rustc-link-search=native=/usr/local/cuda/lib64/stub");
// println!("cargo:rustc-link-lib=cuda");
}