Skip to content

HowTo: Compile CUDA with nvcc, and link to Rust

Notifications You must be signed in to change notification settings

termoshtt/link_cuda_kernel

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HowTo: Compile CUDA with nvcc, and link to Rust through FFI

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

About

HowTo: Compile CUDA with nvcc, and link to Rust

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published