Skip to content

Commit 004ee32

Browse files
committed
[native] Add stage id to arbitration priority for exec::Task
1 parent cd9f2dd commit 004ee32

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

presto-native-execution/presto_cpp/main/TaskManager.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -523,7 +523,9 @@ std::unique_ptr<TaskInfo> TaskManager::createOrUpdateTaskImpl(
523523
planFragment,
524524
prestoTask->id.id(),
525525
std::move(queryCtx),
526-
exec::Task::ExecutionMode::kParallel);
526+
exec::Task::ExecutionMode::kParallel,
527+
static_cast<exec::Consumer>(nullptr),
528+
prestoTask->id.stageId());
527529
// TODO: move spill directory creation inside velox task execution
528530
// whenever spilling is triggered. It will reduce the unnecessary file
529531
// operations on remote storage.

0 commit comments

Comments
 (0)