From: Georgi Gerganov <redacted>
Date: Sun, 8 Sep 2024 06:57:57 +0000 (+0300)
Subject: metal : update support condition for im2col + fix warning (llama/0)
X-Git-Tag: upstream/1.7.4~424
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=03a6fae484ec9c1ff8051daddff331ad8dcb7918;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp

metal : update support condition for im2col + fix warning (llama/0)
---

diff --git a/ggml/src/ggml-metal.m b/ggml/src/ggml-metal.m
index 91b5e61b..f04e5af7 100644
--- a/ggml/src/ggml-metal.m
+++ b/ggml/src/ggml-metal.m
@@ -799,8 +799,9 @@ static bool ggml_metal_supports_op(const struct ggml_backend_metal_context * ctx
             return ctx->support_simdgroup_reduction;
         case GGML_OP_NORM:
         case GGML_OP_ROPE:
-        case GGML_OP_IM2COL:
             return true;
+        case GGML_OP_IM2COL:
+            return op->src[0]->type == GGML_TYPE_F16;
         case GGML_OP_POOL_1D:
         case GGML_OP_POOL_2D:
             return false;