From: Georgi Gerganov <redacted>
Date: Fri, 8 Dec 2023 11:39:32 +0000 (+0200)
Subject: metal : fix soft_max kernel src1 argument (#1602)
X-Git-Tag: upstream/1.7.4~1211
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=7bc4d22337c2ae5538e2e6d89aabe157c2b12eb3;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp

metal : fix soft_max kernel src1 argument (#1602)
---

diff --git a/ggml-metal.m b/ggml-metal.m
index f9bd69dc..df2983f7 100644
--- a/ggml-metal.m
+++ b/ggml-metal.m
@@ -1193,7 +1193,9 @@ void ggml_metal_graph_compute(
                             const float scale = ((float *) dst->op_params)[0];
 
                             [encoder setBuffer:id_src0 offset:offs_src0   atIndex:0];
-                            [encoder setBuffer:id_src1 offset:offs_src1   atIndex:1];
+                            if (id_src1) {
+                                [encoder setBuffer:id_src1 offset:offs_src1   atIndex:1];
+                            }
                             [encoder setBuffer:id_dst  offset:offs_dst    atIndex:2];
                             [encoder setBytes:&ne00  length:sizeof(ne00)  atIndex:3];
                             [encoder setBytes:&ne01  length:sizeof(ne01)  atIndex:4];