From b333b06772c89d96aacb5490d6a219fba7c09cc6 Mon Sep 17 00:00:00 2001 From: Mitja Felicijan Date: Thu, 12 Feb 2026 20:57:17 +0100 Subject: Engage! --- .../ggml/src/ggml-vulkan/vulkan-shaders/relu.comp | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 llama.cpp/ggml/src/ggml-vulkan/vulkan-shaders/relu.comp (limited to 'llama.cpp/ggml/src/ggml-vulkan/vulkan-shaders/relu.comp') diff --git a/llama.cpp/ggml/src/ggml-vulkan/vulkan-shaders/relu.comp b/llama.cpp/ggml/src/ggml-vulkan/vulkan-shaders/relu.comp new file mode 100644 index 0000000..5725cef --- /dev/null +++ b/llama.cpp/ggml/src/ggml-vulkan/vulkan-shaders/relu.comp @@ -0,0 +1,21 @@ +#version 450 + +#include "generic_head.glsl" +#include "types.glsl" + +#extension GL_EXT_control_flow_attributes : enable + +layout(local_size_x = 512, local_size_y = 1, local_size_z = 1) in; + +layout (binding = 0) readonly buffer X {A_TYPE data_a[];}; +layout (binding = 1) writeonly buffer D {D_TYPE data_d[];}; + +void main() { + const uint i = gl_GlobalInvocationID.z * 262144 + gl_GlobalInvocationID.y * 512 + gl_GlobalInvocationID.x; + + if (i >= p.KX) { + return; + } + + data_d[i] = D_TYPE(max(float(data_a[i]), 0)); +} -- cgit v1.2.3