#cuda #kernel #interface #source #automatic #interact #build

build bindgen_cuda

Bindgen like interface to build cuda kernels to interact with within Rust

6 releases

0.1.5 Apr 1, 2024
0.1.4 Feb 6, 2024
0.1.3 Jan 30, 2024

#216 in Algorithms

Download history 183/week @ 2024-01-05 230/week @ 2024-01-12 264/week @ 2024-01-19 387/week @ 2024-01-26 1144/week @ 2024-02-02 1006/week @ 2024-02-09 1466/week @ 2024-02-16 1266/week @ 2024-02-23 1177/week @ 2024-03-01 1167/week @ 2024-03-08 1316/week @ 2024-03-15 2030/week @ 2024-03-22 2076/week @ 2024-03-29 1141/week @ 2024-04-05

6,750 downloads per month
Used in 34 crates (5 directly)

MIT license

24KB
435 lines

Bindgen Cuda

Latest version Documentation License

Similar crate than bindgen in philosophy. It will help create automatic bindgen to cuda kernels source files and make them easier to use directly from Rust.

PTX inclusion

Let's say you have a file

src/cuda.cu

__global__ void cuda_hello(){
    printf("Hello World from GPU!\n");
}

You can add bindgen_cuda as a build dependency:

cargo add --build bindgen_cuda

And then create this build.rs

fn main() {
    let builder = bindgen_cuda::Builder::default();
    let bindings = builder.build_ptx().unwrap();
    bindings.write("src/lib.rs");
}

This will create a src file containing the following code:

pub const CUDA: &str = include_str!(concat!(env!("OUT_DIR"), "/cuda.ptx"));

You can then use the PTX directly in your rust code with a library like cudarc.

Raw cuda calls

Alternatively you can build a static library that you can link against in build.rs in order to call cuda directly with the c code.

src/cuda.cu

__global__ void cuda_hello(){
    printf("Hello World from GPU!\n");
}

int run() {
    cuda_hello<<<1,1>>>(); 
    return 0;
}

Then write the build.rs:

fn main() {
    let builder = bindgen_cuda::Builder::default();
    builder.build_lib("libcuda.a");
    println!("cargo:rustc-link-lib=cuda");
}

Which you can then interface through FFI in src/lib.rs:

extern "C" {
    fn cuda_hello();
}
fn main(){
    unsafe{ cuda_hello();}
}

Dependencies

~1.5MB
~28K SLoC