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

ggml-alloc : correctly check mmap return value for errors

Browse source
This commit is contained in:
slaren 2023-09-08 03:33:02 +02:00
parent 6336d834ec
commit b5b8ff9f51
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
ggml-alloc.c
View file

@ -316,7 +316,11 @@ static void * alloc_vmem(size_t size) {
#if defined(_WIN32) #if defined(_WIN32)
return VirtualAlloc(NULL, size, MEM_RESERVE, PAGE_NOACCESS); return VirtualAlloc(NULL, size, MEM_RESERVE, PAGE_NOACCESS);
#elif defined(_POSIX_MAPPED_FILES) #elif defined(_POSIX_MAPPED_FILES)
return mmap(NULL, size, PROT_NONE, MAP_PRIVATE | MAP_ANON, -1, 0); void * ptr = mmap(NULL, size, PROT_NONE, MAP_PRIVATE | MAP_ANON, -1, 0);
if (ptr == MAP_FAILED) {
return NULL;
}
return ptr;
#else #else
// use a fixed address for other platforms // use a fixed address for other platforms
uintptr_t base_addr = (uintptr_t)-size - 0x100; uintptr_t base_addr = (uintptr_t)-size - 0x100;