gguf : fix comparison (#715)

ggml-ci
ggerganov · Jan 29, 2024 · 475cbad · 475cbad
1 parent b2a5c34
commit 475cbad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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);