hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-09 12:56:46 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

properly handle a forwarded task that gets forwarded back (#6271)

Browse source
This commit is contained in:
Zhijun Fu 2019-12-05 13:37:52 +08:00 • committed by Hao Chen
parent fa98694dd0
commit 7611e484ec
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/ray/raylet/node_manager.cc
View file

@ -2814,6 +2814,12 @@ void NodeManager::ForwardTask(
client->ForwardTask(request, [this, on_error, task, task_id, node_id]( client->ForwardTask(request, [this, on_error, task, task_id, node_id](
Status status, const rpc::ForwardTaskReply &reply) { Status status, const rpc::ForwardTaskReply &reply) {
if (local_queues_.HasTask(task_id)) {
// It must have been forwarded back to us if it's in the queue again
// so just return here.
return;
}
if (status.ok()) { if (status.ok()) {
const auto &spec = task.GetTaskSpecification(); const auto &spec = task.GetTaskSpecification();
// Mark as forwarded so that the task and its lineage are not // Mark as forwarded so that the task and its lineage are not