-
Notifications
You must be signed in to change notification settings - Fork 1
/
build.rs
71 lines (57 loc) · 2.15 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
#[cfg(feature = "cuda")]
extern crate bindgen;
#[cfg(feature = "cuda")]
use std::env;
#[cfg(feature = "cuda")]
use std::path::PathBuf;
#[cfg(feature = "cuda")]
pub fn link_cuda() {
println!("cargo:rerun-if-changed=cluster.hpp");
println!("cargo:rerun-if-changed=cluster.cpp");
println!("cargo:rerun-if-changed=common.hpp");
println!("cargo:rerun-if-changed=common.cpp");
println!("cargo:rerun-if-changed=passthrough.cu");
println!("cargo:rerun-if-changed=passthrough.hpp");
println!("cargo:rerun-if-changed=wrapper.hpp");
let bindings = bindgen::Builder::default()
.header("wrapper.hpp")
.parse_callbacks(Box::new(bindgen::CargoCallbacks::new()))
.generate()
.expect("Unable to generate bindings");
let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
bindings
.write_to_file(out_path.join("bindings.rs"))
.expect("Couldn't write bindings!");
let cuda_include_dir = if cfg!(target_arch = "aarch64") {
"/usr/local/cuda/targets/aarch64-linux/include"
} else {
"/usr/local/cuda/include"
};
cc::Build::new()
.cuda(true)
.cpp_link_stdlib("stdc++")
.files(["cluster.cpp", "common.cpp", "passthrough.cu"])
.include(cuda_include_dir)
.flag_if_supported("-O3")
.compile("cupcl");
println!("cargo:rustc-link-lib=cupcl");
if cfg!(target_arch = "aarch64") {
println!("cargo:rustc-link-search=native=/usr/local/cuda/targets/aarch64-linux/lib");
} else {
println!("cargo:rustc-link-search=native=/usr/local/cuda/lib64");
}
println!("cargo:rustc-link-lib=cudart");
println!("cargo:rustc-link-lib=stdc++");
println!("cargo:rustc-link-search=native=/usr/local/lib");
println!("cargo:rustc-link-lib=cudacluster");
println!("cargo:rustc-link-lib=cudafilter");
}
fn main() {
let cpu_feature_enabled = cfg!(feature = "cpu");
let cuda_feature_enabled = cfg!(feature = "cuda");
if cpu_feature_enabled && cuda_feature_enabled {
panic!("The features 'cpu' and 'cuda' cannot be enabled at the same time.");
}
#[cfg(feature = "cuda")]
link_cuda()
}