vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixing potential NULL dereference in GGML allocator

Browse source
This commit is contained in:
tpltnt 2023-10-10 18:47:37 +02:00
parent 7e60415bc9
commit aebe027100
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
ggml-alloc.c
View file

@ -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_backend_buffer * buffer = ggml_backend_cpu_buffer_from_ptr(NULL, data, size);
struct ggml_allocr * alloc = (struct ggml_allocr *)malloc(sizeof(struct ggml_allocr)); 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){ *alloc = (struct ggml_allocr){
/*.buffer = */ buffer, /*.buffer = */ buffer,