tqcq/llama.cpp
0
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-08-15 12:42:40 -04:00
Code Issues Packages Projects Releases Wiki Activity

ggml-alloc : correctly check mmap return value for errors (#3075)

Browse Source
This commit is contained in:
slaren
2023-09-08 04:04:56 +02:00
committed by GitHub
parent 7f412dab9c
commit ebc96086af
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;