10 lines
258 B
Rust
10 lines
258 B
Rust
|
fn main() {
|
||
|
|
cc::Build::new()
|
||
|
|
.cuda(true)
|
||
|
|
.flag("-cudart=shared")
|
||
|
|
.file("src/c/test.cu")
|
||
|
|
.compile("libtest.a");
|
||
|
|
println!("cargo:rustc-link-search=native=/opt/cuda/lib64");
|
||
|
|
println!("cargo:rustc-link-lib=cudart");
|
||
|
|
}
|