From cfb98641828d79d7ffba94f74b970aed133a520b Mon Sep 17 00:00:00 2001 From: Daniel Bevenius Date: Sun, 11 Aug 2024 10:11:16 +0200 Subject: [PATCH] squash! ggml : move rope type enum to ggml.h Add GGML_ROPE_TYPE_NEOX to rope_common.comp. --- ggml/src/kompute-shaders/rope_common.comp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ggml/src/kompute-shaders/rope_common.comp b/ggml/src/kompute-shaders/rope_common.comp index 7b9394cb2..df4702896 100644 --- a/ggml/src/kompute-shaders/rope_common.comp +++ b/ggml/src/kompute-shaders/rope_common.comp @@ -1,5 +1,7 @@ #include "common.comp" +#define GGML_ROPE_TYPE_NEOX 2 + // TODO: use a local size of 32 or more (Metal uses 1024) layout(local_size_x = 1) in;