(arrow-datafusion) branch main updated: GroupedHashAggregateStream should register spillable consumer (#8002)

[alamb]

This is an automated email from the ASF dual-hosted git repository.

alamb pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/arrow-datafusion.git


The following commit(s) were added to refs/heads/main by this push:
     new 5634cce35f GroupedHashAggregateStream should register spillable 
consumer (#8002)
5634cce35f is described below

commit 5634cce35f6ceb13cf7367b91de9e50d751e442a
Author: Marko Milenković <milenkov...@users.noreply.github.com>
AuthorDate: Wed Nov 1 20:33:28 2023 +0000

    GroupedHashAggregateStream should register spillable consumer (#8002)
---
 datafusion/physical-plan/src/aggregates/row_hash.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/datafusion/physical-plan/src/aggregates/row_hash.rs 
b/datafusion/physical-plan/src/aggregates/row_hash.rs
index 7b66088584..f1eece4315 100644
--- a/datafusion/physical-plan/src/aggregates/row_hash.rs
+++ b/datafusion/physical-plan/src/aggregates/row_hash.rs
@@ -328,7 +328,9 @@ impl GroupedHashAggregateStream {
             .collect();
 
         let name = format!("GroupedHashAggregateStream[{partition}]");
-        let reservation = 
MemoryConsumer::new(name).register(context.memory_pool());
+        let reservation = MemoryConsumer::new(name)
+            .with_can_spill(true)
+            .register(context.memory_pool());
 
         let group_ordering = agg
             .aggregation_ordering