/// With c++17 some of this functionality can be simplified.
#if defined(__linux__) && !defined(__ANDROID__)
-void* aligned_ttmem_alloc(size_t allocSize, void** mem) {
+void* aligned_ttmem_alloc(size_t allocSize, void*& mem) {
constexpr size_t alignment = 2 * 1024 * 1024; // assumed 2MB page sizes
size_t size = ((allocSize + alignment - 1) / alignment) * alignment; // multiple of alignment
- *mem = aligned_alloc(alignment, size);
- madvise(*mem, allocSize, MADV_HUGEPAGE);
- return *mem;
+ mem = aligned_alloc(alignment, size);
+ madvise(mem, allocSize, MADV_HUGEPAGE);
+ return mem;
}
#else
-void* aligned_ttmem_alloc(size_t allocSize, void** mem) {
+void* aligned_ttmem_alloc(size_t allocSize, void*& mem) {
constexpr size_t alignment = 64; // assumed cache line size
size_t size = allocSize + alignment - 1; // allocate some extra space
- *mem = malloc(size);
- void* ret = reinterpret_cast<void*>((uintptr_t(*mem) + alignment - 1) & ~uintptr_t(alignment - 1));
+ mem = malloc(size);
+ void* ret = reinterpret_cast<void*>((uintptr_t(mem) + alignment - 1) & ~uintptr_t(alignment - 1));
return ret;
}