1#extension GL_EXT_control_flow_attributes : require
 2#extension GL_EXT_shader_16bit_storage : require
 3
 4layout (push_constant) uniform parameter
 5{
 6    uint M;
 7    uint K;
 8    uint stride_a;
 9    uint stride_b;
10    uint nel;
11} p;
12
13#include "types.glsl"