1  #extension GL_EXT_control_flow_attributes : require
 2  #extension GL_EXT_shader_16bit_storage : require
 3  
 4  layout (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.comp"