Fix partition output flush

The output size was incorrectly estimated in certain cases due to a bug. This was causing flush to happen either too early or too late depending on a query shape.
arhimondr · Mar 30, 2024 · d10cc45 · d10cc45
1 parent d3503ce
commit d10cc45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/velox/exec/PartitionedOutput.cpp b/velox/exec/PartitionedOutput.cpp
@@ -43,7 +43,7 @@ BlockingReason Destination::advance(
   // Collect rows to serialize.
   bool shouldFlush = false;
   while (rowIdx_ < rows_.size() && !shouldFlush) {
-    bytesInCurrent_ += sizes[rowIdx_];
+    bytesInCurrent_ += sizes[rows_[rowIdx_]];
     ++rowIdx_;
     ++rowsInCurrent_;
     shouldFlush =