Accel: GPGPU Framework for Rust

CUDA-based GPGPU framework for Rust

Features

• Compile PTX Kernel from Rust using NVPTX backend of LLVM (demonstrated in japaric/nvptx)
• proc-macro-attribute-based approach like futures-await
• Simple memory management using Unified Memory

Sub Crates

• accel-derive: Define procedual macro #[kernel]
• accel-core: Support crate for writing GPU kernel
• cuda-sys: Rust binding to CUDA Driver/Runtime APIs

Pre-requirements

• Install CUDA
• Install Rust using rustup.rs
  • accel-derive uses rustup toolchain command.
• Install xargo:

cargo install xargo

Example

#![feature(proc_macro)]

extern crate accel;
extern crate accel_derive;

use accel_derive::kernel;
use accel::*;

#[kernel]
pub unsafe fn add(a: *const f64, b: *const f64, c: *mut f64, n: usize) {
    let i = accel_core::index();
    if (i as usize) < n {
        *c.offset(i) = *a.offset(i) + *b.offset(i);
    }
}

fn main() {
    let n = 8;
    let mut a = UVec::new(n).unwrap();
    let mut b = UVec::new(n).unwrap();
    let mut c = UVec::new(n).unwrap();

    for i in 0..n {
        a[i] = i as f64;
        b[i] = 2.0 * i as f64;
    }
    println!("a = {:?}", a.as_slice());
    println!("b = {:?}", b.as_slice());

    let grid = Grid::x(64);
    let block = Block::x(64);
    add(grid, block, a.as_ptr(), b.as_ptr(), c.as_mut_ptr(), n);

    device::sync().unwrap();
    println!("c = {:?}", c.as_slice());
}

Licence

MIT-License