From 6731796b026910190086ff02191bfa00420d5c4e Mon Sep 17 00:00:00 2001 From: Iwan Kawrakow Date: Sun, 3 Sep 2023 13:14:32 +0300 Subject: [PATCH] Fix bug intriduced in PR #2959 --- ggml-metal.metal | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ggml-metal.metal b/ggml-metal.metal index 3fa311b40..bbd6d7ed6 100644 --- a/ggml-metal.metal +++ b/ggml-metal.metal @@ -552,7 +552,7 @@ kernel void kernel_mul_mat_f16_f32_1row( } float all_sum = simd_sum(sumf); if (tiisg == 0) { - for (int i = 4*(ne00/4); i < ne00; ++i) sumf += (float) x[i] * y[i]; + for (int i = 4*(ne00/4); i < ne00; ++i) all_sum += (float) x[i] * y[i]; dst[im*ne1*ne0 + r1*ne0 + r0] = all_sum; } } @@ -625,7 +625,7 @@ kernel void kernel_mul_mat_f16_f32( float all_sum = simd_sum(sumf); if (tiisg == 0) { - for (int i = 4*(ne00/4); i < ne00; ++i) sumf += (float) x[i] * y[i]; + for (int i = 4*(ne00/4); i < ne00; ++i) all_sum += (float) x[i] * y[i]; dst[im*ne1*ne0 + r1*ne0 + r0] = all_sum; } }