From 765fab636a31a12e2d053dab45e8651f1799966e Mon Sep 17 00:00:00 2001 From: klosax <131523366+klosax@users.noreply.github.com> Date: Fri, 28 Jul 2023 22:31:14 +0200 Subject: [PATCH] ggml.c : add gguf_find_key --- ggml.c | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/ggml.c b/ggml.c index 96c7ebd34..d402daa67 100644 --- a/ggml.c +++ b/ggml.c @@ -18745,6 +18745,21 @@ int gguf_get_n_kv(struct gguf_context * ctx) { return ctx->header.n_kv; } +int gguf_find_key(struct gguf_context * ctx, const char * key) { + // return -1 if key not found + const int n_kv = gguf_get_n_kv(ctx); + int keyfound = -1; + + for (int i = 0; i < n_kv; ++i) { + if (strcmp(key, gguf_get_key(ctx, i)) == 0) { + keyfound = i; + break; + } + } + + return keyfound; +} + const char * gguf_get_key(struct gguf_context * ctx, int i) { return ctx->header.kv[i].key.data; }