Skip to content

Commit

Permalink
[GLUTEN-7243][VL] A follow-up fix for #7748 (#7935)
Browse files Browse the repository at this point in the history
(cherry picked from commit 61033b7)
  • Loading branch information
zhztheplayer authored and liuxiang71 committed Nov 15, 2024
1 parent c875d6c commit 88cca21
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions cpp/velox/operators/plannodes/RowVectorStream.h
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ class RowVectorStream {
// As of now, non-zero running threads usually happens when:
// 1. Task A spills task B;
// 2. Task A trys to grow buffers created by task B, during which spill is requested on task A again.
facebook::velox::exec::SuspendedSection(driverCtx_->driver);
facebook::velox::exec::SuspendedSection ss(driverCtx_->driver);
hasNext = iterator_->hasNext();
}
if (!hasNext) {
Expand All @@ -64,7 +64,7 @@ class RowVectorStream {
{
// We are leaving Velox task execution and are probably entering Spark code through JNI. Suspend the current
// driver to make the current task open to spilling.
facebook::velox::exec::SuspendedSection(driverCtx_->driver);
facebook::velox::exec::SuspendedSection ss(driverCtx_->driver);
cb = iterator_->next();
}
const std::shared_ptr<VeloxColumnarBatch>& vb = VeloxColumnarBatch::from(pool_, cb);
Expand Down

0 comments on commit 88cca21

Please sign in to comment.