File: spv.subgroupShuffle.comp

package info (click to toggle)
glslang 16.2.0-1
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 51,720 kB
  • sloc: cpp: 92,305; yacc: 4,320; sh: 603; python: 305; ansic: 94; javascript: 74; makefile: 17
file content (68 lines) | stat: -rw-r--r-- 3,710 bytes parent folder | download | duplicates (26)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#version 450

#extension GL_KHR_shader_subgroup_shuffle: 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;
    dvec4 d4;
} data[4];

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

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

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

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

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

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

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

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

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

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

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