From 2abb6c722599b543457e6f9d73a6fc266e66d620 Mon Sep 17 00:00:00 2001 From: slaren Date: Sat, 30 Mar 2024 11:42:28 +0100 Subject: [PATCH] Update ggml-metal.m Co-authored-by: Georgi Gerganov --- ggml-metal.m | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ggml-metal.m b/ggml-metal.m index 336f8e740..69ffb39b7 100644 --- a/ggml-metal.m +++ b/ggml-metal.m @@ -1734,9 +1734,9 @@ static enum ggml_status ggml_metal_graph_compute( // some Metal matrix data types require aligned pointers // ref: https://developer.apple.com/metal/Metal-Shading-Language-Specification.pdf (Table 2.5) - switch (src2->type) { - case GGML_TYPE_F32: GGML_ASSERT(nb21 % 16 == 0); break; - case GGML_TYPE_F16: GGML_ASSERT(nb21 % 8 == 0); break; + switch (src0->type) { + case GGML_TYPE_F32: GGML_ASSERT(nb01 % 16 == 0); break; + case GGML_TYPE_F16: GGML_ASSERT(nb01 % 8 == 0); break; default: break; }