From aebe027100a9fab82d9e35a1c3dbe50f730dc32c Mon Sep 17 00:00:00 2001 From: tpltnt <1172976+tpltnt@users.noreply.github.com> Date: Tue, 10 Oct 2023 18:47:37 +0200 Subject: [PATCH] fixing potential NULL dereference in GGML allocator --- ggml-alloc.c | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/ggml-alloc.c b/ggml-alloc.c index 34eba3f83..f9a4ce221 100644 --- a/ggml-alloc.c +++ b/ggml-alloc.c @@ -271,6 +271,10 @@ struct ggml_allocr * ggml_allocr_new(void * data, size_t size, size_t alignment) struct ggml_backend_buffer * buffer = ggml_backend_cpu_buffer_from_ptr(NULL, data, size); struct ggml_allocr * alloc = (struct ggml_allocr *)malloc(sizeof(struct ggml_allocr)); + if (NULL == alloc) { + fprintf(stderr, "could not retrieve memory for GGML allocator.\n"); + exit(1); + } *alloc = (struct ggml_allocr){ /*.buffer = */ buffer,