From e8b00e29415df61145bc1d76b63b118da003f34f Mon Sep 17 00:00:00 2001
From: Georgi Gerganov <ggerganov@gmail.com>
Date: Thu, 8 Feb 2024 16:39:38 +0200
Subject: [PATCH] metal : fix NSG1 > 1

---
 ggml-metal.metal | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ggml-metal.metal b/ggml-metal.metal
index f795f5386..dba9935af 100644
--- a/ggml-metal.metal
+++ b/ggml-metal.metal
@@ -4980,7 +4980,7 @@ void kernel_mul_mm2_impl(
 
     for (int is = 0; is < NSG1; is++) {
         const int i1 = i11 + is*8;
-        const int nstore = MIN(8*NSG1, ne1 - i1) * (8*NSG0);
+        const int nstore = MIN(8, ne1 - i1) * (8*NSG0);
 
         for (int i = tiisg; i < nstore; i += NW) {
             const int ic = i%(8*NSG0);