Skip to content

Commit

Permalink
exclude previous operator's time out of firstBatchHeuristic (NVIDIA#1…
Browse files Browse the repository at this point in the history
…1794)

Signed-off-by: Hongbin Ma (Mahone) <[email protected]>
  • Loading branch information
binmahone authored Dec 3, 2024
1 parent cb31afb commit 738c8e3
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -2091,9 +2091,9 @@ class DynamicGpuPartialAggregateIterator(
helper: AggHelper): (Iterator[ColumnarBatch], Boolean) = {
// we need to decide if we are going to sort the data or not, so the very
// first thing we need to do is get a batch and make a choice.
val cb = cbIter.next()
withResource(new NvtxWithMetrics("dynamic sort heuristic", NvtxColor.BLUE,
metrics.opTime, metrics.heuristicTime)) { _ =>
val cb = cbIter.next()
lazy val estimatedGrowthAfterAgg: Double = closeOnExcept(cb) { cb =>
val numRows = cb.numRows()
val cardinality = estimateCardinality(cb)
Expand Down

0 comments on commit 738c8e3

Please sign in to comment.