gpgpu 0.2.0

Simple WIP GPGPU framework built on top of wgpu
Documentation 
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

struct Dims {
    x: u32;
    y: u32;
};

struct Array {
    data: [[stride(4)]] array<i32>;
};

[[group(0), binding(0)]] var<uniform> dims: Dims;   // Array dimensions

[[group(1), binding(0)]] var<storage, read> a: Array;           
[[group(1), binding(1)]] var<storage, read> b: Array;           
[[group(1), binding(2)]] var<storage, read_write> c: Array;  

// fn main_fn_1([[builtin(global_invocation_id)]] global_id: vec3<u32>) {
//     let idx = global_id.x; 
//     c.data[idx] = a.data[idx] + b.data[idx];
// }

[[stage(compute), workgroup_size(32, 32)]]
fn main_fn_2([[builtin(global_invocation_id)]] global_id: vec3<u32>) {
    let id = (global_id.x * dims.x) + global_id.y;
    c.data[id] = a.data[id] + b.data[id];
}