File tree Expand file tree Collapse file tree
datafusion/physical-plan/src/aggregates Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -41,7 +41,7 @@ use arrow::datatypes::SchemaRef;
4141use datafusion_common:: { internal_err, DataFusionError , Result } ;
4242use datafusion_execution:: disk_manager:: RefCountedTempFile ;
4343use datafusion_execution:: memory_pool:: proxy:: VecAllocExt ;
44- use datafusion_execution:: memory_pool:: { MemoryConsumer , MemoryReservation } ;
44+ use datafusion_execution:: memory_pool:: { MemoryConsumer , MemoryLimit , MemoryReservation } ;
4545use datafusion_execution:: TaskContext ;
4646use datafusion_expr:: { EmitTo , GroupsAccumulator } ;
4747use datafusion_physical_expr:: expressions:: Column ;
@@ -683,7 +683,7 @@ fn maybe_enable_blocked_groups(
683683 . enable_aggregation_blocked_groups
684684 || !matches ! ( group_ordering, GroupOrdering :: None )
685685 || accumulators. is_empty ( )
686- || context. runtime_env ( ) . disk_manager . tmp_files_enabled ( )
686+ || matches ! ( context. memory_pool ( ) . memory_limit ( ) , MemoryLimit :: Infinite )
687687 {
688688 return Ok ( false ) ;
689689 }
You can’t perform that action at this time.
0 commit comments