From e3cafcd24013c8932362a847c75a97f133c99287 Mon Sep 17 00:00:00 2001
From: Rong Ma <rong.ma@intel.com>
Date: Tue, 21 Jun 2022 23:41:39 +0800
Subject: [PATCH] TaskCursor use global memory pool (#214)

---
 cpp/velox/compute/VeloxPlanConverter.cc | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/cpp/velox/compute/VeloxPlanConverter.cc b/cpp/velox/compute/VeloxPlanConverter.cc
index a1eb2f6bb527..a785284b30bc 100644
--- a/cpp/velox/compute/VeloxPlanConverter.cc
+++ b/cpp/velox/compute/VeloxPlanConverter.cc
@@ -381,6 +381,13 @@ class VeloxPlanConverter::WholeStageResIterFirstStage : public WholeStageResIter
       splits_.emplace_back(exec::Split(folly::copy(connectorSplit), -1));
     }
     params_.planNode = planNode;
+    params_.queryCtx = std::make_shared<core::QueryCtx>(
+        pool,
+        std::make_shared<folly::CPUThreadPoolExecutor>(
+            std::thread::hardware_concurrency()),
+        std::make_shared<core::MemConfig>(),
+        std::unordered_map<std::string, std::shared_ptr<Config>>{},
+        memory::MappedMemory::getInstance());
     cursor_ = std::make_unique<test::TaskCursor>(params_);
     addSplits_ = [&](Task* task) {
       if (noMoreSplits_) {
@@ -411,6 +418,13 @@ class VeloxPlanConverter::WholeStageResIterMiddleStage : public WholeStageResIte
                                const bool fakeArrowOutput)
       : WholeStageResIter(pool, planNode) {
     params_.planNode = planNode;
+    params_.queryCtx = std::make_shared<core::QueryCtx>(
+        pool,
+        std::make_shared<folly::CPUThreadPoolExecutor>(
+            std::thread::hardware_concurrency()),
+        std::make_shared<core::MemConfig>(),
+        std::unordered_map<std::string, std::shared_ptr<Config>>{},
+        memory::MappedMemory::getInstance());
     cursor_ = std::make_unique<test::TaskCursor>(params_);
     addSplits_ = [&](Task* task) {
       if (noMoreSplits_) {