You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
992 B
33 lines
992 B
4 months ago
|
#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
|
||
|
|
}
|