fixup

apache · Aug 12, 2024 · 33acd2d · 33acd2d
1 parent 7d91fa7
commit 33acd2d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/velox/tests/MemoryManagerTest.cc b/cpp/velox/tests/MemoryManagerTest.cc
@@ -392,7 +392,7 @@ TEST_F(MultiMemoryManagerTest, spill) {
 
   for (auto& vmm : vmms) {
     assertCapacitiesMatch(tmm, vmms);
-    vmm->getMemoryManager()->arbitrator()->shrinkCapacity(allocateSize);
+    vmm->getMemoryManager()->arbitrator()->shrinkCapacity(allocateSize * numAllocations);
     assertCapacitiesMatch(tmm, vmms);
   }