Skip to content
Snippets Groups Projects

GPU backend

Merged prathi3 requested to merge gpu-cg into main
1 file
+ 0
25
Compare changes
  • Side-by-side
  • Inline
+ 0
25
#[entry]
fn main<m, n : usize>() -> i32[m, n, 64, 64] {
let res : i32[m, n, 64, 64];
for bi = 0 to m {
for bj = 0 to n {
let tile : i32[64, 64];
for ti = 0 to 64 {
for tj = 0 to 64 {
tile[ti, tj] = (ti as i32) + (tj as i32) + (bi as i32) + (bj as i32);
}
}
for si = 1 to 63 {
for sj = 1 to 63 {
tile[si, sj] = tile[si-1, sj-1] + tile[si+1, sj+1];
}
}
for ri = 0 to 64 {
for rj = 0 to 64 {
res[bi, bj, ri, rj] = tile[ri, rj];
}
}
}
}
return res;
}
Loading