Build kernel.cu (copied from CUDA sample) into libvector_add.a in build.rs:
extern crate cc;
fn main() {
cc::Build::new()
.cuda(true)
.flag("-cudart=shared")
.flag("-gencode")
.flag("arch=compute_61,code=sm_61")
.file("kernel.cu")
.compile("libvector_add.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: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");
}and link this host code into Rust executable:
#[link(name = "vector_add", kind = "static")]
extern "C" {
fn vectorAdd_main();
}
fn main() {
unsafe {
vectorAdd_main();
}
}/** CUDA Kernel Device code */
__global__ void vectorAdd(const float *A, const float *B, float *C, int numElements) {
int i = blockDim.x * blockIdx.x + threadIdx.x;
if (i < numElements)
{
C[i] = A[i] + B[i];
}
}
/** Host main routine */
extern "C" { // To avoid demangle
int vectorAdd_main (void) {
/* call kernel in CUDA/C++ way */
}
} // extern C