]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
metal : add f16 type support to leaky relu (#25981)
authorIlia Ilmer <redacted>
Thu, 23 Jul 2026 03:45:46 +0000 (23:45 -0400)
committerGitHub <redacted>
Thu, 23 Jul 2026 03:45:46 +0000 (11:45 +0800)
ggml/src/ggml-metal/ggml-metal-device.m

index 80e47f2c2f89b9728a8804b9bec3489423be7e83..4edd77c6f267a79d95be0a85d35ac4f2cc69b5bd 100644 (file)
@@ -1218,8 +1218,9 @@ bool ggml_metal_device_supports_op(ggml_metal_device_t dev, const struct ggml_te
                    (ggml_get_op_params_i32(op, 4) == 0) && (ggml_get_op_params_i32(op, 6) == 0);
         case GGML_OP_PAD_REFLECT_1D:
         case GGML_OP_TIMESTEP_EMBEDDING:
-        case GGML_OP_LEAKY_RELU:
             return op->src[0]->type == GGML_TYPE_F32;
+        case GGML_OP_LEAKY_RELU:
+            return op->src[0]->type == GGML_TYPE_F32 || op->src[0]->type == GGML_TYPE_F16;
         case GGML_OP_ARGSORT:
         case GGML_OP_TOP_K:
         case GGML_OP_ARANGE: