xref: /aosp_15_r20/external/angle/third_party/glslang/src/Test/glsl.es320.subgroupBallotNeg.comp (revision 8975f5c5ed3d1c378011245431ada316dfb6f244)
1#version 450
2
3#extension GL_KHR_shader_subgroup_ballot: enable
4
5layout (local_size_x = 8, local_size_y = 8, local_size_z = 1) in;
6
7layout(binding = 0) buffer Buffers
8{
9    vec4  f4;
10    ivec4 i4;
11    uvec4 u4;
12} data[4];
13
14void main()
15{
16    uint invocation = (gl_SubgroupInvocationID + gl_SubgroupSize) % 4u;
17
18    uvec4 relMask = gl_SubgroupEqMask +
19                       gl_SubgroupGeMask +
20                       gl_SubgroupGtMask +
21                       gl_SubgroupLeMask +
22                       gl_SubgroupLtMask;
23
24    uvec4 result = subgroupBallot(true);
25
26    data[0].u4.x = subgroupBallotBitCount(result);
27    data[0].u4.y = subgroupBallotBitExtract(result, 0) ? 1u : 0u;
28    data[0].u4.z = subgroupBallotInclusiveBitCount(result) + subgroupBallotExclusiveBitCount(result);
29    data[0].u4.w = subgroupBallotFindLSB(result) + subgroupBallotFindMSB(result);
30
31    data[1].f4.x   = subgroupBroadcast(data[0].f4.x,    invocation);  // ERROR: not constant
32}
33