123456789101112131415161718192021222324252627282930313233 |
- #version 450
- #extension GL_KHR_shader_subgroup_ballot: enable
- layout (local_size_x = 8, local_size_y = 8, local_size_z = 1) in;
- layout(binding = 0) buffer Buffers
- {
- vec4 f4;
- ivec4 i4;
- uvec4 u4;
- } data[4];
- void main()
- {
- uint invocation = (gl_SubgroupInvocationID + gl_SubgroupSize) % 4u;
- uvec4 relMask = gl_SubgroupEqMask +
- gl_SubgroupGeMask +
- gl_SubgroupGtMask +
- gl_SubgroupLeMask +
- gl_SubgroupLtMask;
- uvec4 result = subgroupBallot(true);
- data[0].u4.x = subgroupBallotBitCount(result);
- data[0].u4.y = subgroupBallotBitExtract(result, 0) ? 1u : 0u;
- data[0].u4.z = subgroupBallotInclusiveBitCount(result) + subgroupBallotExclusiveBitCount(result);
- data[0].u4.w = subgroupBallotFindLSB(result) + subgroupBallotFindMSB(result);
- data[1].f4.x = subgroupBroadcast(data[0].f4.x, invocation); // ERROR: not constant
- }
|