kiper220/mimalloc
1
0
Fork 0
mirror of https://github.com/microsoft/mimalloc.git synced 2025-05-21 06:19:31 +03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge 83228bb973 into 43e5cd2671

Browse source
This commit is contained in:
David CARLIER 2022-01-03 04:19:18 +09:00 • committed by GitHub
commit b24aed9ef8
WARNING! Although there is a key with this ID in the database it does not verify this commit! This commit is SUSPICIOUS.
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 33 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
src/os.c
View file

@ -98,6 +98,11 @@ static size_t os_alloc_granularity = 4096;
// if non-zero, use large page allocation
static size_t large_os_page_size = 0;
#if defined(MADV_HUGEPAGE)
// only linux supports the THP's notion.
static bool os_transparent_huge_pages = false;
#endif
// is memory overcommit allowed?
// set dynamically in _mi_os_init (and if true we use MAP_NORESERVE)
static bool os_overcommit = true;
@ -238,6 +243,30 @@ void _mi_os_init() {
#else // generic unix
static void os_detect_transparent_huge_pages(void) {
#if defined(MADV_HUGEPAGE)
int fd = -1;
size_t i;
static const char *paths[] = {
"/sys/kernel/mm/transparent_hugepage/enabled",
"/sys/kernel/mm/redhat_transparent_hugepage/enabled",
};
for (i = 0; i < sizeof(paths) / sizeof(paths[0]); i ++) {
fd = open(paths[i], O_RDONLY);
if (fd != -1) break;
}
if (fd < 0) return;
char buf[128];
ssize_t nread = read(fd, &buf, sizeof(buf));
close(fd);
if (nread >= 1) {
if (strstr(buf, "[madvise]")) os_transparent_huge_pages = true;
}
#endif
}
static void os_detect_overcommit(void) {
#if defined(__linux__)
int fd = open("/proc/sys/vm/overcommit_memory", O_RDONLY);
@ -270,6 +299,7 @@ void _mi_os_init() {
}
large_os_page_size = 2*MI_MiB; // TODO: can we query the OS for this?
os_detect_overcommit();
os_detect_transparent_huge_pages();
}
#endif
@ -588,7 +618,7 @@ static void* mi_unix_mmap(void* addr, size_t size, size_t try_alignment, int pro
// in that case -- in particular for our large regions (in `memory.c`).
// However, some systems only allow THP if called with explicit `madvise`, so
// when large OS pages are enabled for mimalloc, we call `madvise` anyways.
if (allow_large && use_large_os_page(size, try_alignment)) {
if (os_transparent_huge_pages && allow_large && use_large_os_page(size, try_alignment)) {
if (madvise(p, size, MADV_HUGEPAGE) == 0) {
*is_large = true; // possibly
};