From b7af83267744d159efdb5442e67e7303d2ae61f4 Mon Sep 17 00:00:00 2001
From: Hongze Zhang <hongze.zhang@intel.com>
Date: Wed, 29 Nov 2023 12:19:25 +0800
Subject: [PATCH] [VL] Fix assertion failures when mixing use of partial
 aggregation spilling and flushing  (#3872)

---
 cpp/velox/compute/WholeStageResultIterator.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cpp/velox/compute/WholeStageResultIterator.cc b/cpp/velox/compute/WholeStageResultIterator.cc
index 9f2a3e4ab6e7..ae6a81ad9dc7 100644
--- a/cpp/velox/compute/WholeStageResultIterator.cc
+++ b/cpp/velox/compute/WholeStageResultIterator.cc
@@ -356,6 +356,7 @@ std::unordered_map<std::string, std::string> WholeStageResultIterator::getQueryC
     }
     configs[velox::core::QueryConfig::kAggregationSpillEnabled] =
         getConfigValue(confMap_, kAggregationSpillEnabled, "true");
+    configs[velox::core::QueryConfig::kPartialAggregationSpillEnabled] = "true";
     configs[velox::core::QueryConfig::kJoinSpillEnabled] = getConfigValue(confMap_, kJoinSpillEnabled, "true");
     configs[velox::core::QueryConfig::kOrderBySpillEnabled] = getConfigValue(confMap_, kOrderBySpillEnabled, "true");
     configs[velox::core::QueryConfig::kAggregationSpillMemoryThreshold] =