From f07ec1bc17f4253d2a0434996a760b134e252796 Mon Sep 17 00:00:00 2001
From: gn64 <yukikaze.jp@gmail.com>
Date: Tue, 3 Dec 2024 06:17:05 +0900
Subject: [PATCH] Update soft_max.comp

This change prevents a division by zero error when p.KY is 0.
---
 ggml/src/ggml-vulkan/vulkan-shaders/soft_max.comp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ggml/src/ggml-vulkan/vulkan-shaders/soft_max.comp b/ggml/src/ggml-vulkan/vulkan-shaders/soft_max.comp
index f9727679ec5..d372025b9f8 100644
--- a/ggml/src/ggml-vulkan/vulkan-shaders/soft_max.comp
+++ b/ggml/src/ggml-vulkan/vulkan-shaders/soft_max.comp
@@ -32,7 +32,7 @@ shared FLOAT_TYPE vals[BLOCK_SIZE];
 void soft_max(uint num_iters) {
     const uint tid = gl_LocalInvocationID.x;
     const uint rowx = gl_WorkGroupID.z * 262144 + gl_WorkGroupID.y * 512 + gl_WorkGroupID.x;
-    const uint rowy = rowx % p.KY;
+    const uint rowy = (p.KY > 0) ? (rowx % p.KY) : 0;
 
     if (rowx >= p.nrows_x) {
         return;