debug window

FelixYBW · Jul 3, 2024 · fd92d7f · fd92d7f
1 parent 72c609d
commit fd92d7f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/velox/exec/Window.cpp b/velox/exec/Window.cpp
@@ -718,6 +718,11 @@ RowVectorPtr Window::getOutput() {
       currentPartition_->clearOutputRows(numResultRows.second);
     }
   }
+  numBatches_++;
+  if (numBatches_ % 10 == 0 )
+  {
+    std::cout << this->pool()->root()->treeMemoryUsage() << std::endl;
+  }
 
   return numResultRows.first < numOutputRows
       ? std::dynamic_pointer_cast<RowVector>(

diff --git a/velox/exec/Window.h b/velox/exec/Window.h
@@ -231,6 +231,7 @@ class Window : public Operator {
   // computePeerBuffers they are saved here.
   vector_size_t peerStartRow_ = 0;
   vector_size_t peerEndRow_ = 0;
+  uint64_t numBatches_ = 0;
 };
 
 } // namespace facebook::velox::exec