Free

mohsaka · May 24, 2024 · d6b0787 · d6b0787
1 parent 80c6abb
commit d6b0787
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/velox/common/memory/MemoryAllocator.cpp b/velox/common/memory/MemoryAllocator.cpp
@@ -271,6 +271,7 @@ bool MemoryAllocator::allocateContiguous(
     success = cache()->makeSpace(
         pagesToAcquire(numPages, numCollateralPages),
         [&](Allocation& acquired) {
+          freeNonContiguous(acquired);
           unmap(acquired.numPages());
           return allocateContiguousWithoutRetry(
               numPages, collateral, allocation, maxPages);
@@ -306,6 +307,7 @@ bool MemoryAllocator::growContiguous(
   } else {
     success = cache()->makeSpace(increment, [&](Allocation& acquired) {
       freeNonContiguous(acquired);
+      unmap(acquired.numPages());
       return growContiguousWithoutRetry(increment, allocation);
     });
   }