vulkan : fix soft_max.comp division by zero (#2633)

This change prevents a division by zero error when p.KY is 0.
2024-12-18 20:27:53 +00:00 · 2024-12-16 19:34:38 +09:00 · 2024-12-16 19:34:38 +09:00 · c4aed6831e
commit c4aed6831e
parent 199579652e
1 changed files with 1 additions and 1 deletions
--- 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;