wgpu/naga/tests/out/wgsl/subgroup-operations.wgsl
Alexander Meißner ea77d5674d
Subgroup Operations (#5301)
Co-authored-by: Jacob Hughes <j@distanthills.org>
Co-authored-by: Connor Fitzgerald <connorwadefitzgerald@gmail.com>
Co-authored-by: atlas dostal <rodol@rivalrebels.com>
2024-04-17 15:25:52 -04:00

32 lines
1.6 KiB
WebGPU Shading Language

struct Structure {
@builtin(num_subgroups) num_subgroups: u32,
@builtin(subgroup_size) subgroup_size: u32,
}
@compute @workgroup_size(1, 1, 1)
fn main(sizes: Structure, @builtin(subgroup_id) subgroup_id: u32, @builtin(subgroup_invocation_id) subgroup_invocation_id: u32) {
subgroupBarrier();
let _e7 = subgroupBallot(((subgroup_invocation_id & 1u) == 1u));
let _e8 = subgroupBallot();
let _e11 = subgroupAll((subgroup_invocation_id != 0u));
let _e14 = subgroupAny((subgroup_invocation_id == 0u));
let _e15 = subgroupAdd(subgroup_invocation_id);
let _e16 = subgroupMul(subgroup_invocation_id);
let _e17 = subgroupMin(subgroup_invocation_id);
let _e18 = subgroupMax(subgroup_invocation_id);
let _e19 = subgroupAnd(subgroup_invocation_id);
let _e20 = subgroupOr(subgroup_invocation_id);
let _e21 = subgroupXor(subgroup_invocation_id);
let _e22 = subgroupExclusiveAdd(subgroup_invocation_id);
let _e23 = subgroupExclusiveMul(subgroup_invocation_id);
let _e24 = subgroupInclusiveAdd(subgroup_invocation_id);
let _e25 = subgroupInclusiveMul(subgroup_invocation_id);
let _e26 = subgroupBroadcastFirst(subgroup_invocation_id);
let _e28 = subgroupBroadcast(subgroup_invocation_id, 4u);
let _e33 = subgroupShuffle(subgroup_invocation_id, ((sizes.subgroup_size - 1u) - subgroup_invocation_id));
let _e35 = subgroupShuffleDown(subgroup_invocation_id, 1u);
let _e37 = subgroupShuffleUp(subgroup_invocation_id, 1u);
let _e41 = subgroupShuffleXor(subgroup_invocation_id, (sizes.subgroup_size - 1u));
return;
}