external/vulkancts/data/vulkan/amber/subgroup_uniform_control_flow/large/subgroup_reconverge15.amber

#!amber

SHADER compute test GLSL TARGET_ENV spv1.3
#version 450 core
#extension GL_KHR_shader_subgroup_basic : enable
#extension GL_KHR_shader_subgroup_ballot : enable
#extension GL_EXT_subgroup_uniform_control_flow : enable
layout(local_size_x = 128, local_size_y = 2, local_size_z = 1) in;

layout(set=0, binding=0) coherent buffer A { uint a[]; } a;
layout(set=0, binding=1) coherent buffer B { uint b[]; } b;
layout(set=0, binding=2) coherent buffer C { uint c[]; } c;
layout(set=0, binding=3) volatile buffer D { uint d[]; } d;

void main()
[[subgroup_uniform_control_flow]]
{
  uint idx = gl_SubgroupID * gl_SubgroupSize + gl_SubgroupInvocationID;
  switch (a.a[idx]) {
    case 1:
      atomicAdd(d.d[idx], 1);
      break;
    case 3:
      return;
    default:
      switch (c.c[idx]) {
        case 0:
        case 1:
          atomicExchange(d.d[0], gl_SubgroupID);
          break;
        case 2:
          c.c[idx]++;
          break;
        default:
          atomicAdd(d.d[1], 32);
          break;
      }
      b.b[idx] = subgroupElect() ? 1 : 0;
      break;
  }
}
END

SHADER compute fill GLSL TARGET_ENV spv1.3
#version 450 core
#extension GL_KHR_shader_subgroup_basic : enable
#extension GL_KHR_shader_subgroup_ballot : enable
layout(local_size_x = 128, local_size_y = 2, local_size_z = 1) in;

layout(set=0, binding=0) coherent buffer A { uint a[]; } a;
layout(set=0, binding=1) coherent buffer C { uint c[]; } c;
layout(set=0, binding=2) coherent buffer COMPARE { uint x[]; } compare;

void main()
{
  uint idx = gl_SubgroupID * gl_SubgroupSize + gl_SubgroupInvocationID;
  uint cmp_val = subgroupElect() ? 1 : 0;
  if (gl_SubgroupID % 2 != 0) {
    cmp_val = 4;
  }
  compare.x[idx] = cmp_val;

  a.a[idx] = gl_SubgroupID % 4;

  c.c[idx] = gl_SubgroupID;
}
END

BUFFER a DATA_TYPE uint32 SIZE 256 FILL 0
BUFFER b DATA_TYPE uint32 SIZE 256 FILL 4
BUFFER c DATA_TYPE uint32 SIZE 256 FILL 4
BUFFER d DATA_TYPE uint32 SIZE 256 FILL 0
BUFFER compare DATA_TYPE uint32 SIZE 256 FILL 4

PIPELINE compute fill_pipe
  ATTACH fill
  BIND BUFFER a AS storage DESCRIPTOR_SET 0 BINDING 0
  BIND BUFFER c AS storage DESCRIPTOR_SET 0 BINDING 1
  BIND BUFFER compare AS storage DESCRIPTOR_SET 0 BINDING 2
END

PIPELINE compute test_pipe
  ATTACH test
  BIND BUFFER a AS storage DESCRIPTOR_SET 0 BINDING 0
  BIND BUFFER b AS storage DESCRIPTOR_SET 0 BINDING 1
  BIND BUFFER c AS storage DESCRIPTOR_SET 0 BINDING 2
  BIND BUFFER d AS storage DESCRIPTOR_SET 0 BINDING 3
END

RUN fill_pipe 1 1 1
RUN test_pipe 1 1 1

EXPECT compare IDX 0 EQ 1 0 0 0
EXPECT b EQ_BUFFER compare