yum-mirror/glslang

git clone https://git.yummers.dev/yum-mirror/glslang

Jeff BolzImplement GL_EXT_long_vector (#4132)b937eae5

main
branchesmainall refs →
2.4 KiB103 linesraw
1#version 450 core
2#extension GL_KHR_memory_scope_semantics : enable
3#extension GL_EXT_long_vector : enable
4#extension GL_EXT_shader_explicit_arithmetic_types : enable
5#extension GL_EXT_buffer_reference : enable
6#extension GL_EXT_nonuniform_qualifier : enable
7
8layout (local_size_x = 64, local_size_y = 1, local_size_z = 1) in;
9
10layout(set = 0, binding = 0) buffer MatrixBuf {
11    float16_t matrixData[];
12} matrixBuf;
13
14layout(set = 0, binding = 0) buffer MatrixBuf2 {
15    int8_t matrixData[];
16} matrixBuf2;
17
18layout(constant_id = 0) const int matrixLayout = gl_CooperativeVectorMatrixLayoutColumnMajorNV;
19
20void main()
21{
22    vector<float, 5> v = vector<float, 5>(0.0);
23
24    vec3 x = vec3(1.0), y = vec3(2.0);
25    vector<float, 5> v2 = vector<float, 5>(x, y);
26
27    const vec3 x2 = vec3(1.0), y2 = vec3(2.0);
28    vector<float, 5> v3 = vector<float, 5>(x2, y2);
29
30    vec3 x3 = vec3(4, 5, 6), y3 = vec3(7, 8, 9);
31
32    vector<float, 5> v4 = vector<float, 5>(x3.zyx, y3.zzy);
33
34    vector<float16_t, 5> v5 = vector<float16_t, 5>(v4);
35
36    vector<float, 5> v6;
37
38    v6 = v + v2;
39    v6 = v - v2;
40    v6 = v * v2;
41    v6 = v / v2;
42
43    v6 += v;
44    v6 -= v;
45    v6 *= v;
46    v6 /= v;
47
48    float f;
49    v6 *= f;
50    v6 *= 5.0;
51
52    int len = v6.length();
53
54    v6[0] = f;
55    v6[4] = 5.0;
56
57    f += (v += v)[4];
58    v6 = v;
59
60    vector<float, 20> v7;
61    vector<float16_t, 2> f162 = vector<float16_t, 2>(v7.length());
62
63    float16_t f16;
64    vector<float16_t, 1> f161 = vector<float16_t, 1>(f16);
65
66    vector<float16_t, 7> v11, v12, v13;
67
68    v11 = max(v11, v12);
69    v11 = min(v11, v12);
70    v11 = step(v11, v12);
71    v11 = clamp(v11, v12, v13);
72    v11 = exp(v11);
73    v11 = log(v11);
74    v11 = tanh(v11);
75    v11 = atan(v11);
76    v11 = fma(v11, v12, v13);
77
78    f += (v += v)[len];
79
80    v13 = vector<float16_t, 7>(v13);
81    f16 = float16_t(v13);
82    vector<float16_t, 2> vecA;
83    vecA = exp(vecA * vector<float16_t, 2>(0.0625));
84
85}
86
87vector<float16_t, 1> foo(vector<float16_t, 1> x) { return x; }
88vector<float16_t, 2> foo(vector<float16_t, 2> x) { return x; }
89
90vec4 funcvec4(vec4 a) { return a; }
91vector<float, 4> funcvector4(vector<float, 4> a) { return a; }
92
93void testfuncs()
94{
95    vec4 a;
96    vector<float, 4> b;
97    a = b;
98    b = a;
99    funcvec4(b);
100    funcvector4(a);
101    vector<float16_t, 1> x;
102    x = foo(x);
103}