Test/spv.shaderBallot.comp

   1 #version 450\r
   2 \r
   3 #extension GL_ARB_gpu_shader_int64: enable\r
   4 #extension GL_ARB_shader_ballot: enable\r
   5 \r
   6 layout (local_size_x = 8, local_size_y = 8, local_size_z = 1) in;\r
   7 \r
   8 layout(binding = 0) buffer Buffers\r
   9 {\r
  10     vec4  f4;\r
  11     ivec4 i4;\r
  12     uvec4 u4; \r
  13 } data[4];\r
  14 \r
  15 void main()\r
  16 {\r
  17     uint invocation = (gl_SubGroupInvocationARB + gl_SubGroupSizeARB) % 4;\r
  18 \r
  19     uint64_t relMask = gl_SubGroupEqMaskARB +\r
  20                        gl_SubGroupGeMaskARB +\r
  21                        gl_SubGroupGtMaskARB +\r
  22                        gl_SubGroupLeMaskARB +\r
  23                        gl_SubGroupLtMaskARB;\r
  24 \r
  25     if (relMask == ballotARB(true))\r
  26     {\r
  27         data[invocation].f4.x   = readInvocationARB(data[0].f4.x,    invocation);\r
  28         data[invocation].f4.xy  = readInvocationARB(data[1].f4.xy,   invocation);\r
  29         data[invocation].f4.xyz = readInvocationARB(data[2].f4.xyz,  invocation);\r
  30         data[invocation].f4     = readInvocationARB(data[3].f4,      invocation);\r
  31 \r
  32         data[invocation].i4.x   = readInvocationARB(data[0].i4.x,    invocation);\r
  33         data[invocation].i4.xy  = readInvocationARB(data[1].i4.xy,   invocation);\r
  34         data[invocation].i4.xyz = readInvocationARB(data[2].i4.xyz,  invocation);\r
  35         data[invocation].i4     = readInvocationARB(data[3].i4,      invocation);\r
  36 \r
  37         data[invocation].u4.x   = readInvocationARB(data[0].u4.x,    invocation);\r
  38         data[invocation].u4.xy  = readInvocationARB(data[1].u4.xy,   invocation);\r
  39         data[invocation].u4.xyz = readInvocationARB(data[2].u4.xyz,  invocation);\r
  40         data[invocation].u4     = readInvocationARB(data[3].u4,      invocation);\r
  41     }\r
  42     else\r
  43     {\r
  44         data[invocation].f4.x   = readFirstInvocationARB(data[0].f4.x);\r
  45         data[invocation].f4.xy  = readFirstInvocationARB(data[1].f4.xy);\r
  46         data[invocation].f4.xyz = readFirstInvocationARB(data[2].f4.xyz);\r
  47         data[invocation].f4     = readFirstInvocationARB(data[3].f4);\r
  48 \r
  49         data[invocation].i4.x   = readFirstInvocationARB(data[0].i4.x);\r
  50         data[invocation].i4.xy  = readFirstInvocationARB(data[1].i4.xy);\r
  51         data[invocation].i4.xyz = readFirstInvocationARB(data[2].i4.xyz);\r
  52         data[invocation].i4     = readFirstInvocationARB(data[3].i4);\r
  53 \r
  54         data[invocation].u4.x   = readFirstInvocationARB(data[0].u4.x);\r
  55         data[invocation].u4.xy  = readFirstInvocationARB(data[1].u4.xy);\r
  56         data[invocation].u4.xyz = readFirstInvocationARB(data[2].u4.xyz);\r
  57         data[invocation].u4     = readFirstInvocationARB(data[3].u4);\r
  58     }\r
  59 }