nvptx 0.2.3

Rust to PTX compiler using LLVM/NVPTX target
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

extern crate nvptx;

use nvptx::*;

const GPU_CODE: &'static str = r#"
#![feature(abi_ptx)]
#![no_std]
extern crate accel_core;
#[no_mangle]
pub unsafe extern "ptx-kernel" 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);
    }
}
"#;

#[test]
fn compile_tmp() {
    let dri = Driver::new().unwrap();
    let core = manifest::Crate::new("accel-core", "0.2.0-alpha");
    manifest::generate(dri.path(), &[core]).unwrap();
    let ptx = dri.compile_str(GPU_CODE).unwrap();
    println!("PTX = {:?}", ptx);
}

#[test]
fn compile_path() {
    let dri = Driver::new().unwrap();
    let core = manifest::Crate::new("accel-core", "0.2.0-alpha");
    manifest::generate(dri.path(), &[core]).unwrap();
    let ptx = dri.compile_str(GPU_CODE).unwrap();
    println!("PTX = {:?}", ptx);
}