glslang/Test/glsl.es320.subgroupShuffleRelative.comp

#version 320 es

#extension GL_KHR_shader_subgroup_shuffle_relative: 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;

    data[0].f4.x   = subgroupShuffleUp(data[0].f4.x,    invocation);
    data[0].f4.xy  = subgroupShuffleUp(data[1].f4.xy,   invocation);
    data[0].f4.xyz = subgroupShuffleUp(data[2].f4.xyz,  invocation);
    data[0].f4     = subgroupShuffleUp(data[3].f4,      invocation);

    data[0].i4.x   = subgroupShuffleUp(data[0].i4.x,    invocation);
    data[0].i4.xy  = subgroupShuffleUp(data[1].i4.xy,   invocation);
    data[0].i4.xyz = subgroupShuffleUp(data[2].i4.xyz,  invocation);
    data[0].i4     = subgroupShuffleUp(data[3].i4,      invocation);

    data[1].u4.x   = subgroupShuffleUp(data[0].u4.x,    invocation);
    data[1].u4.xy  = subgroupShuffleUp(data[1].u4.xy,   invocation);
    data[1].u4.xyz = subgroupShuffleUp(data[2].u4.xyz,  invocation);
    data[1].u4     = subgroupShuffleUp(data[3].u4,      invocation);

    data[1].i4.x   =   int(subgroupShuffleUp(data[0].i4.x < 0,                   invocation));
    data[1].i4.xy  = ivec2(subgroupShuffleUp(lessThan(data[1].i4.xy, ivec2(0)),  invocation));
    data[1].i4.xyz = ivec3(subgroupShuffleUp(lessThan(data[1].i4.xyz, ivec3(0)), invocation));
    data[1].i4     = ivec4(subgroupShuffleUp(lessThan(data[1].i4, ivec4(0)),     invocation));

    data[2].f4.x   = subgroupShuffleDown(data[0].f4.x,    invocation);
    data[2].f4.xy  = subgroupShuffleDown(data[1].f4.xy,   invocation);
    data[2].f4.xyz = subgroupShuffleDown(data[2].f4.xyz,  invocation);
    data[2].f4     = subgroupShuffleDown(data[3].f4,      invocation);

    data[2].i4.x   = subgroupShuffleDown(data[0].i4.x,    invocation);
    data[2].i4.xy  = subgroupShuffleDown(data[1].i4.xy,   invocation);
    data[2].i4.xyz = subgroupShuffleDown(data[2].i4.xyz,  invocation);
    data[2].i4     = subgroupShuffleDown(data[3].i4,      invocation);

    data[3].u4.x   = subgroupShuffleDown(data[0].u4.x,    invocation);
    data[3].u4.xy  = subgroupShuffleDown(data[1].u4.xy,   invocation);
    data[3].u4.xyz = subgroupShuffleDown(data[2].u4.xyz,  invocation);
    data[3].u4     = subgroupShuffleDown(data[3].u4,      invocation);

    data[3].i4.x   =   int(subgroupShuffleDown(data[0].i4.x < 0,                   invocation));
    data[3].i4.xy  = ivec2(subgroupShuffleDown(lessThan(data[1].i4.xy, ivec2(0)),  invocation));
    data[3].i4.xyz = ivec3(subgroupShuffleDown(lessThan(data[1].i4.xyz, ivec3(0)), invocation));
    data[3].i4     = ivec4(subgroupShuffleDown(lessThan(data[1].i4, ivec4(0)),     invocation));
}