diff --git a/cpp/velox/compute/WholeStageResultIterator.cc b/cpp/velox/compute/WholeStageResultIterator.cc index eb700c6489ec..498583ca2145 100644 --- a/cpp/velox/compute/WholeStageResultIterator.cc +++ b/cpp/velox/compute/WholeStageResultIterator.cc @@ -209,29 +209,6 @@ std::shared_ptr WholeStageResultIterator::next() { return std::make_shared(vector); } -namespace { -class SuspendedSection { - public: - SuspendedSection() { - reclaimer_->enterArbitration(); - } - - virtual ~SuspendedSection() { - reclaimer_->leaveArbitration(); - } - - // singleton - SuspendedSection(const SuspendedSection&) = delete; - SuspendedSection(SuspendedSection&&) = delete; - SuspendedSection& operator=(const SuspendedSection&) = delete; - SuspendedSection& operator=(SuspendedSection&&) = delete; - - private: - // We only use suspension APIs in exec::MemoryReclaimer. - std::unique_ptr reclaimer_{velox::exec::MemoryReclaimer::create()}; -}; -} // namespace - int64_t WholeStageResultIterator::spillFixedSize(int64_t size) { auto pool = memoryManager_->getAggregateMemoryPool(); std::string poolName{pool->root()->name() + "/" + pool->name()}; @@ -241,9 +218,6 @@ int64_t WholeStageResultIterator::spillFixedSize(int64_t size) { if (spillStrategy_ == "auto") { int64_t remaining = size - shrunken; LOG(INFO) << logPrefix << "Trying to request spilling for " << remaining << " bytes..."; - // suspend the driver when we are on it - SuspendedSection suspender; - velox::exec::MemoryReclaimer::Stats status; auto* mm = memoryManager_->getMemoryManager(); uint64_t spilledOut = mm->arbitrator()->shrinkCapacity(remaining); // this conducts spilling LOG(INFO) << logPrefix << "Successfully spilled out " << spilledOut << " bytes.";