dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

rm wgsl

This commit is contained in:
louisfd 2024-02-29 23:58:41 -05:00
parent f13dbebdf1
commit fc5ed13d2d
1 changed files with 0 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
crates/burn-wgpu/src/template/bool_cast.wgsl
View File

@ -1,21 +0,0 @@
@group(0)
@binding(0)
var<storage, read> input: array<u32>;
@group(0)
@binding(1)
var<storage, read_write> output: array<{{ output_elem }}>;
@compute
@workgroup_size({{ workgroup_size_x }}, {{ workgroup_size_y }}, 1)
fn main(
@builtin(global_invocation_id) global_id: vec3<u32>,
@builtin(num_workgroups) num_workgroups: vec3<u32>,
) {
let id = global_id.y * (num_workgroups.x * {{ workgroup_size_x }}u) + global_id.x;
if (input[id] > 0u){
output[id] = {{ output_elem }}(1);
} else {
output[id] = {{ output_elem }}(0);
}
}