unroll

Author: adamant-pwn

cp-algo/util/big_alloc.hpp

@@ -26,18 +26,28 @@ namespace cp_algo {
 
 #if CP_ALGO_USE_MMAP
         [[nodiscard]] T* allocate(std::size_t n) {
-            auto *raw = base::allocate(n);
-            if(n * sizeof(T) >= (1 << 20)) {
-                madvise(raw, n, MADV_HUGEPAGE);
-                madvise(raw, n, MADV_POPULATE_WRITE);
+            if(n * sizeof(T) < 1024 * 1024) {
+                return base::allocate(n);
             }
-            return raw;
+            n *= sizeof(T);
+            void* raw = mmap(nullptr, n,
+                            PROT_READ | PROT_WRITE,
+                            MAP_PRIVATE | MAP_ANONYMOUS,
+                            -1, 0);
+            madvise(raw, n, MADV_HUGEPAGE);
+            madvise(raw, n, MADV_POPULATE_WRITE);
+            return static_cast<T*>(raw);
         }
 #endif
 
 #if CP_ALGO_USE_MMAP
         void deallocate(T* p, std::size_t n) noexcept {
-            return base::deallocate(p, n);
+            if(n * sizeof(T) < 1024 * 1024) {
+                return base::deallocate(p, n);
+            }
+            if(p) {
+                munmap(p, n * sizeof(T));
+            }
         }
 #endif
     };