From: Georgi Gerganov <redacted>
Date: Mon, 29 Jan 2024 19:08:18 +0000 (+0200)
Subject: gguf : fix comparison (#715)
X-Git-Tag: upstream/0.0.1642~1021
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=475cbad5c1c834e31e26a2283bc1413181644360;p=pkg%2Fggml%2Fsources%2Fggml

gguf : fix comparison (#715)

ggml-ci
---

diff --git a/src/ggml.c b/src/ggml.c
index 954041e2..e6dce1c4 100644
--- a/src/ggml.c
+++ b/src/ggml.c
@@ -19421,7 +19421,7 @@ struct gguf_context * gguf_init_from_file(const char * fname, struct gguf_init_p
                             case GGUF_TYPE_BOOL:
                                 {
                                     // prevent from integer overflow in the malloc below
-                                    if (kv->value.arr.n < SIZE_MAX/gguf_type_size(kv->value.arr.type)) {
+                                    if (kv->value.arr.n >= SIZE_MAX/gguf_type_size(kv->value.arr.type)) {
                                         fprintf(stderr, "%s: array size is too large (%" PRIu64 ")\n", __func__, kv->value.arr.n);
                                         fclose(file);
                                         gguf_free(ctx);
@@ -19435,7 +19435,7 @@ struct gguf_context * gguf_init_from_file(const char * fname, struct gguf_init_p
                             case GGUF_TYPE_STRING:
                                 {
                                     // prevent from integer overflow in the malloc below
-                                    if (kv->value.arr.n < SIZE_MAX/sizeof(struct gguf_str)) {
+                                    if (kv->value.arr.n >= SIZE_MAX/sizeof(struct gguf_str)) {
                                         fprintf(stderr, "%s: array size is too large (%" PRIu64 ")\n", __func__, kv->value.arr.n);
                                         fclose(file);
                                         gguf_free(ctx);