]> git.djapps.eu Git - pkg/ggml/sources/whisper.cpp/commitdiff
metal : update support condition for im2col + fix warning (llama/0)
authorGeorgi Gerganov <redacted>
Sun, 8 Sep 2024 06:57:57 +0000 (09:57 +0300)
committerGeorgi Gerganov <redacted>
Tue, 24 Sep 2024 16:45:08 +0000 (19:45 +0300)
ggml/src/ggml-metal.m

index 91b5e61b23eadfed8d349a5cd0114c39f07f5285..f04e5af71f9ebdc57dd39fe39d9bb4834d00c664 100644 (file)
@@ -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;