From 211d82a800916961ce7c087472de341f1fb965b1 Mon Sep 17 00:00:00 2001 From: Georgi Gerganov Date: Mon, 11 Sep 2023 10:29:34 +0300 Subject: [PATCH] metal : minor (readibility) --- ggml-metal.metal | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/ggml-metal.metal b/ggml-metal.metal index 36a31e29a..f45b1490f 100644 --- a/ggml-metal.metal +++ b/ggml-metal.metal @@ -213,10 +213,9 @@ kernel void kernel_diag_mask_inf_8( dst[i+0] = src0[i+0]; dst[i+1] = src0[i+1]; int64_t i4 = 4*i; - const int64_t i02 = i4/(ne00*ne01); - i4 -= i02*ne00*ne01; - const int64_t i01 = i4/ne00; - const int64_t i00 = i4 - i01*ne00; + const int64_t i02 = i4/(ne00*ne01); i4 -= i02*ne00*ne01; + const int64_t i01 = i4/(ne00); i4 -= i01*ne00; + const int64_t i00 = i4; for (int k = 3; k >= 0; --k) { if (i00 + 4 + k <= n_past + i01) { break; @@ -678,7 +677,6 @@ kernel void kernel_mul_mat_f16_f32_l4( device const half4 * x4 = (device const half4 *) (src0 + r0*nb01 + im/(ne12/ne02)*nb02); for (int r1 = 0; r1 < nrows; ++r1) { - device const float4 * y4 = (device const float4 *) (src1 + r1*nb11 + im*nb12); float sumf = 0;