Avoid unnecessary reclaim operation on non-reclaimable query

facebookincubator · May 7, 2024 · 92c08ae · 92c08ae
1 parent 1fd2bc9
commit 92c08ae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/velox/common/memory/SharedArbitrator.cpp b/velox/common/memory/SharedArbitrator.cpp
@@ -725,7 +725,8 @@ uint64_t SharedArbitrator::reclaim(
     uint64_t targetBytes,
     bool isLocalArbitration) noexcept {
   int64_t bytesToReclaim = std::min<uint64_t>(
-      std::max(targetBytes, memoryPoolTransferCapacity_), pool->capacity());
+      std::max(targetBytes, memoryPoolTransferCapacity_),
+      reclaimableUsedCapacity(*pool));
   if (bytesToReclaim == 0) {
     return 0;
   }