diff --git a/crates/burn-wgpu/src/template/bool_cast.wgsl b/crates/burn-wgpu/src/template/bool_cast.wgsl deleted file mode 100644 index cdf08862e..000000000 --- a/crates/burn-wgpu/src/template/bool_cast.wgsl +++ /dev/null @@ -1,21 +0,0 @@ -@group(0) -@binding(0) -var input: array; - -@group(0) -@binding(1) -var output: array<{{ output_elem }}>; - -@compute -@workgroup_size({{ workgroup_size_x }}, {{ workgroup_size_y }}, 1) -fn main( - @builtin(global_invocation_id) global_id: vec3, - @builtin(num_workgroups) num_workgroups: vec3, -) { - 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); - } -}