7c7836d9d4
* Refactor shaders, extract GLSL code from ggml_vk_generate_shaders.py into vulkan-shaders directory * Improve debug log code * Add memory debug output option * Fix flake8 * Fix unnecessary high llama-3 VRAM use
13 lines
249 B
Text
13 lines
249 B
Text
#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"
|