From af95ae49a385050ae1f491bc62c019b3c32f9016 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Johannes=20G=C3=A4=C3=9Fler?= <johannesg@5d6.de>
Date: Wed, 29 May 2024 19:34:37 +0200
Subject: [PATCH] fix metal tests

---
 ggml-metal.m | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/ggml-metal.m b/ggml-metal.m
index ff9ae55aa..f5e27c6fa 100644
--- a/ggml-metal.m
+++ b/ggml-metal.m
@@ -779,6 +779,12 @@ static bool ggml_metal_supports_op(const struct ggml_metal_context * ctx, const
         case GGML_OP_LEAKY_RELU:
             return true;
         case GGML_OP_FLASH_ATTN_EXT:
+            if (op->src[1]->type != GGML_TYPE_F16) {
+                return false;
+            }
+            if (op->src[2]->type != GGML_TYPE_F16) {
+                return false;
+            }
             if (op->src[0]->ne[0] == 256) {
                 return false;
             }