From d221ff8bc209f464d80e1bc32e4d8ae53a7edf85 Mon Sep 17 00:00:00 2001 From: Stephanie Date: Tue, 5 Feb 2019 15:10:43 -0800 Subject: [PATCH] Only mark tasks as forwarded if they are in the lineage cache --- src/ray/raylet/node_manager.cc | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/ray/raylet/node_manager.cc b/src/ray/raylet/node_manager.cc index 892335a6123b..d5f665364c2e 100644 --- a/src/ray/raylet/node_manager.cc +++ b/src/ray/raylet/node_manager.cc @@ -1948,10 +1948,11 @@ void NodeManager::ForwardTask(const Task &task, const ClientID &node_id, if (!lineage_cache_.RemoveWaitingTask(task_id)) { RAY_LOG(WARNING) << "Task " << task_id << " already removed from the lineage" << " cache. This is most likely due to reconstruction."; + } else { + // Mark as forwarded so that the task and its lineage is not + // re-forwarded in the future to the receiving node. + lineage_cache_.MarkTaskAsForwarded(task_id, node_id); } - // Mark as forwarded so that the task and its lineage is not re-forwarded - // in the future to the receiving node. - lineage_cache_.MarkTaskAsForwarded(task_id, node_id); // Notify the task dependency manager that we are no longer responsible // for executing this task.