fixup

zhztheplayer · Aug 19, 2024 · 05e1e2b · 05e1e2b
1 parent 1c04efd
commit 05e1e2b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/velox/common/memory/MemoryPool.cpp b/velox/common/memory/MemoryPool.cpp
@@ -27,6 +27,11 @@
 
 #include <re2/re2.h>
 
+DEFINE_bool(
+    memory_pool_capacity_transfer_across_tasks,
+    false,
+    "Whether allow to memory capacity transfer between memory pools from different tasks, which might happen in use case like Spark-Gluten");
+
 DECLARE_bool(velox_suppress_memory_capacity_exceeding_error_message);
 
 using facebook::velox::common::testutil::TestValue;

diff --git a/velox/exec/MemoryReclaimer.cpp b/velox/exec/MemoryReclaimer.cpp
@@ -19,11 +19,6 @@
 #include "velox/exec/Driver.h"
 #include "velox/exec/Task.h"
 
-DEFINE_bool(
-    transferred_arbitration_allowed,
-    false,
-    "Whether to allow entering memory arbitration on memory pool that is from another task");
-
 namespace facebook::velox::exec {
 std::unique_ptr<memory::MemoryReclaimer> MemoryReclaimer::create() {
   return std::unique_ptr<memory::MemoryReclaimer>(new MemoryReclaimer());

diff --git a/velox/exec/MemoryReclaimer.h b/velox/exec/MemoryReclaimer.h
@@ -18,7 +18,7 @@
 
 #include "velox/common/memory/MemoryArbitrator.h"
 
-DECLARE_bool(transferred_arbitration_allowed);
+DECLARE_bool(memory_pool_capacity_transfer_across_tasks);
 
 namespace facebook::velox::exec {
 /// Provides the default leaf memory reclaimer implementation for velox task