From cfee0c8140e97b598c18f1bc79ce37ae2a39a4b2 Mon Sep 17 00:00:00 2001
From: Ke <ke.404notfound@gmail.com>
Date: Tue, 2 Apr 2024 14:01:03 -0700
Subject: [PATCH] Switch to estimateCompactSize when reporting outputBytes
 statistics

---
 velox/exec/Driver.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/velox/exec/Driver.cpp b/velox/exec/Driver.cpp
index f2f43c6b451a5..1ed716644b86c 100644
--- a/velox/exec/Driver.cpp
+++ b/velox/exec/Driver.cpp
@@ -590,7 +590,7 @@ StopReason Driver::runInternal(
                 if (ctx_->queryConfig().validateOutputFromOperators()) {
                   validateOperatorResult(intermediateResult, *op);
                 }
-                resultBytes = intermediateResult->estimateFlatSize();
+                resultBytes = intermediateResult->estimateCompactSize();
                 {
                   auto lockedStats = op->stats().wlock();
                   lockedStats->addOutputVector(
@@ -699,7 +699,7 @@ StopReason Driver::runInternal(
               {
                 auto lockedStats = op->stats().wlock();
                 lockedStats->addOutputVector(
-                    result->estimateFlatSize(), result->size());
+                    result->estimateCompactSize(), result->size());
               }
 
               // This code path is used only in single-threaded execution.