From 7c7836d9d4062d6858e3fb337b135c417ccee6ce Mon Sep 17 00:00:00 2001 From: 0cc4m Date: Sun, 16 Jun 2024 07:17:31 +0200 Subject: Vulkan Shader Refactor, Memory Debugging Option (#7947) * 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 --- vulkan-shaders/generic_head.comp | 9 +++++++++ 1 file changed, 9 insertions(+) create mode 100644 vulkan-shaders/generic_head.comp (limited to 'vulkan-shaders/generic_head.comp') diff --git a/vulkan-shaders/generic_head.comp b/vulkan-shaders/generic_head.comp new file mode 100644 index 00000000..66e46ae6 --- /dev/null +++ b/vulkan-shaders/generic_head.comp @@ -0,0 +1,9 @@ +#extension GL_EXT_shader_16bit_storage : require + +layout (push_constant) uniform parameter +{ + uint KX; + uint KY; + float param1; + float param2; +} p; -- cgit v1.2.3