From fc5ed13d2db7459b33a53306154afa85d77cefd3 Mon Sep 17 00:00:00 2001 From: louisfd Date: Thu, 29 Feb 2024 23:58:41 -0500 Subject: [PATCH] rm wgsl --- crates/burn-wgpu/src/template/bool_cast.wgsl | 21 -------------------- 1 file changed, 21 deletions(-) delete mode 100644 crates/burn-wgpu/src/template/bool_cast.wgsl 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); - } -}