diff options
Diffstat (limited to 'cli/tests/testdata/webgpu_computepass_shader.wgsl')
-rw-r--r-- | cli/tests/testdata/webgpu_computepass_shader.wgsl | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/cli/tests/testdata/webgpu_computepass_shader.wgsl b/cli/tests/testdata/webgpu_computepass_shader.wgsl index 9fc79eeba..df541aff5 100644 --- a/cli/tests/testdata/webgpu_computepass_shader.wgsl +++ b/cli/tests/testdata/webgpu_computepass_shader.wgsl @@ -1,9 +1,10 @@ -[[block]] struct PrimeIndices { data: [[stride(4)]] array<u32>; }; // this is used as both input and output for convenience + [[group(0), binding(0)]] var<storage, read_write> v_indices: PrimeIndices; + // The Collatz Conjecture states that for any integer n: // If n is even, n = n/2 // If n is odd, n = 3n+1 @@ -25,12 +26,14 @@ fn collatz_iterations(n_base: u32) -> u32{ if (n >= 1431655765u) { // 0x55555555u return 4294967295u; // 0xffffffffu } + n = 3u * n + 1u; } i = i + 1u; } return i; } + [[stage(compute), workgroup_size(1)]] fn main([[builtin(global_invocation_id)]] global_id: vec3<u32>) { v_indices.data[global_id.x] = collatz_iterations(v_indices.data[global_id.x]); |