squash! ggml : move rope type enum to ggml.h

Add GGML_ROPE_TYPE_NEOX to rope_common.comp.
2024-08-11 10:11:16 +02:00 · 2024-08-11 10:11:16 +02:00 · cfb9864182
commit cfb9864182
parent ecd1c1e6b8
1 changed files with 2 additions and 0 deletions
--- 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;