1 #version 450 2 3 #extension GL_KHR_shader_subgroup_shuffle_relative: enable 4 5 layout (local_size_x = 8, local_size_y = 8, local_size_z = 1) in; 6 7 layout(binding = 0) buffer Buffers 8 { 9 vec4 f4; 10 ivec4 i4; 11 uvec4 u4; 12 dvec4 d4; 13 } data[4]; 14 15 void main() 16 { 17 uint invocation = (gl_SubgroupInvocationID + gl_SubgroupSize) % 4; 18 19 data[invocation].f4.x = subgroupShuffleUp(data[0].f4.x, invocation); 20 data[invocation].f4.xy = subgroupShuffleUp(data[1].f4.xy, invocation); 21 data[invocation].f4.xyz = subgroupShuffleUp(data[2].f4.xyz, invocation); 22 data[invocation].f4 = subgroupShuffleUp(data[3].f4, invocation); 23 24 data[invocation].i4.x = subgroupShuffleUp(data[0].i4.x, invocation); 25 data[invocation].i4.xy = subgroupShuffleUp(data[1].i4.xy, invocation); 26 data[invocation].i4.xyz = subgroupShuffleUp(data[2].i4.xyz, invocation); 27 data[invocation].i4 = subgroupShuffleUp(data[3].i4, invocation); 28 29 data[invocation].u4.x = subgroupShuffleUp(data[0].u4.x, invocation); 30 data[invocation].u4.xy = subgroupShuffleUp(data[1].u4.xy, invocation); 31 data[invocation].u4.xyz = subgroupShuffleUp(data[2].u4.xyz, invocation); 32 data[invocation].u4 = subgroupShuffleUp(data[3].u4, invocation); 33 34 data[invocation].d4.x = subgroupShuffleUp(data[0].d4.x, invocation); 35 data[invocation].d4.xy = subgroupShuffleUp(data[1].d4.xy, invocation); 36 data[invocation].d4.xyz = subgroupShuffleUp(data[2].d4.xyz, invocation); 37 data[invocation].d4 = subgroupShuffleUp(data[3].d4, invocation); 38 39 data[invocation].i4.x = int(subgroupShuffleUp(data[0].i4.x < 0, invocation)); 40 data[invocation].i4.xy = ivec2(subgroupShuffleUp(lessThan(data[1].i4.xy, ivec2(0)), invocation)); 41 data[invocation].i4.xyz = ivec3(subgroupShuffleUp(lessThan(data[1].i4.xyz, ivec3(0)), invocation)); 42 data[invocation].i4 = ivec4(subgroupShuffleUp(lessThan(data[1].i4, ivec4(0)), invocation)); 43 44 data[invocation].f4.x = subgroupShuffleDown(data[0].f4.x, invocation); 45 data[invocation].f4.xy = subgroupShuffleDown(data[1].f4.xy, invocation); 46 data[invocation].f4.xyz = subgroupShuffleDown(data[2].f4.xyz, invocation); 47 data[invocation].f4 = subgroupShuffleDown(data[3].f4, invocation); 48 49 data[invocation].i4.x = subgroupShuffleDown(data[0].i4.x, invocation); 50 data[invocation].i4.xy = subgroupShuffleDown(data[1].i4.xy, invocation); 51 data[invocation].i4.xyz = subgroupShuffleDown(data[2].i4.xyz, invocation); 52 data[invocation].i4 = subgroupShuffleDown(data[3].i4, invocation); 53 54 data[invocation].u4.x = subgroupShuffleDown(data[0].u4.x, invocation); 55 data[invocation].u4.xy = subgroupShuffleDown(data[1].u4.xy, invocation); 56 data[invocation].u4.xyz = subgroupShuffleDown(data[2].u4.xyz, invocation); 57 data[invocation].u4 = subgroupShuffleDown(data[3].u4, invocation); 58 59 data[invocation].d4.x = subgroupShuffleDown(data[0].d4.x, invocation); 60 data[invocation].d4.xy = subgroupShuffleDown(data[1].d4.xy, invocation); 61 data[invocation].d4.xyz = subgroupShuffleDown(data[2].d4.xyz, invocation); 62 data[invocation].d4 = subgroupShuffleDown(data[3].d4, invocation); 63 64 data[invocation].i4.x = int(subgroupShuffleDown(data[0].i4.x < 0, invocation)); 65 data[invocation].i4.xy = ivec2(subgroupShuffleDown(lessThan(data[1].i4.xy, ivec2(0)), invocation)); 66 data[invocation].i4.xyz = ivec3(subgroupShuffleDown(lessThan(data[1].i4.xyz, ivec3(0)), invocation)); 67 data[invocation].i4 = ivec4(subgroupShuffleDown(lessThan(data[1].i4, ivec4(0)), invocation)); 68 } 69
