xref: /third_party/glslang/Test/glsl.es320.subgroupVote.comp

#version 320 es

#extension GL_KHR_shader_subgroup_vote: 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;
    int r;
} data[4];

void main()
{
    uint invocation = (gl_SubgroupInvocationID + gl_SubgroupSize) % 4u;

    if (subgroupAll(data[0].r < 0))
    {
        data[0].r = int(subgroupAllEqual(data[0].f4.x));
        data[0].r = int(subgroupAllEqual(data[1].f4.xy));
        data[0].r = int(subgroupAllEqual(data[2].f4.xyz));
        data[0].r = int(subgroupAllEqual(data[3].f4));

        data[0].r = int(subgroupAllEqual(data[0].i4.x));
        data[0].r = int(subgroupAllEqual(data[1].i4.xy));
        data[0].r = int(subgroupAllEqual(data[2].i4.xyz));
        data[0].r = int(subgroupAllEqual(data[3].i4));

        data[0].r = int(subgroupAllEqual(data[0].u4.x));
        data[0].r = int(subgroupAllEqual(data[1].u4.xy));
        data[0].r = int(subgroupAllEqual(data[2].u4.xyz));
        data[0].r = int(subgroupAllEqual(data[3].u4));
    }
    else if (subgroupAny(data[1].r < 0))
    {
        data[1].r = int(int(subgroupAllEqual(data[0].i4.x < 0)));
        data[1].r = int(ivec2(subgroupAllEqual(lessThan(data[1].i4.xy, ivec2(0)))));
        data[1].r = int(ivec3(subgroupAllEqual(lessThan(data[1].i4.xyz, ivec3(0)))));
        data[1].r = int(ivec4(subgroupAllEqual(lessThan(data[1].i4, ivec4(0)))));
    }
}