memory: wire up mmap huge page support

[fio.git] / memory.c

diff --git a/memory.c b/memory.c
index 82a79bd7175b9a74ca0846a669d767e06d50b84d..b4ee4752487129be62d98a126d305c62fb1c0b0d 100644 (file)
--- a/memory.c
+++ b/memory.c
@@ -121,6 +121,13 @@ static int alloc_mem_mmap(struct thread_data *td, size_t total_mem)
 
        td->mmapfd = 1;
 
+       if (td->o.mem_type == MEM_MMAPHUGE) {
+               unsigned long mask = td->o.hugepage_size - 1;
+
+               flags |= MAP_HUGETLB;
+               total_mem = (total_mem + mask) & ~mask;
+       }
+
        if (td->mmapfile) {
                td->mmapfd = open(td->mmapfile, O_RDWR|O_CREAT, 0644);