yum-mirror/glslang

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

Jeff BolzImplement GL_EXT_shader_64bit_indexing (#4082)e2ab21f2

main
branchesmainall refs →
1.3 KiB34 linesraw
1#version 450 core
2#extension GL_KHR_memory_scope_semantics : enable
3#extension GL_KHR_cooperative_matrix : enable
4#extension GL_NV_cooperative_vector : enable
5#extension GL_EXT_shader_explicit_arithmetic_types : enable
6#extension GL_EXT_buffer_reference : enable
7
8layout (local_size_x = 64, local_size_y = 1, local_size_z = 1) in;
9
10layout(set = 0, binding = 0, buffer_reference) coherent buffer Block {
11    float x[];
12} block;
13
14void main()
15{
16    uint64_t element = 0;
17    coopmat<float16_t, gl_ScopeSubgroup, 16, 16, gl_MatrixUseA> m;
18
19    // errors missing 64bit_indexing
20    coopMatLoad(m, block.x, element, 128, gl_CooperativeMatrixLayoutRowMajor);
21    coopMatStore(m, block.x, element, 128, gl_CooperativeMatrixLayoutRowMajor);
22
23    coopvecNV<float16_t, 2> vec;
24    coopVecMatMulAddNV(vec, vec, 0, block.x, element, 0, block.x, element, 0, 2, 2, 0, false, 0);
25    coopVecMatMulNV(vec, vec, 0, block.x, element, 0, 2, 2, 0, false, 0);
26    coopVecLoadNV(vec, block.x, element);
27    coopVecStoreNV(vec, block.x, element);
28    coopVecOuterProductAccumulateNV(vec, vec, block.x, element, 0, 0, 0);
29    coopVecReduceSumAccumulateNV(vec, block.x, element);
30
31    float y = block.x[element];
32    int32_t len = block.x.length();
33    int64_t len64 = block.x.length64();
34}