ggml/src/vulkan-shaders/dequant_head.comp


1
2
3
4
5
6
7
8
9
10
11
12
13

#extension GL_EXT_control_flow_attributes : require
#extension GL_EXT_shader_16bit_storage : require

layout (push_constant) uniform parameter
{
    uint M;
    uint K;
    uint stride_a;
    uint stride_b;
    uint nel;
} p;

#include "types.comp"