From 85b8b2a39597e43be55a24339354a9a41c6baba2 Mon Sep 17 00:00:00 2001
From: Alexey Tumanov <atumanov@gmail.com>
Date: Mon, 6 Aug 2018 13:08:11 -0700
Subject: [PATCH] mark all remaining placeable tasks pending with task
 dependency manager (#2528)

---
 src/ray/raylet/node_manager.cc | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/ray/raylet/node_manager.cc b/src/ray/raylet/node_manager.cc
index 1072e5a6f..2be170a46 100644
--- a/src/ray/raylet/node_manager.cc
+++ b/src/ray/raylet/node_manager.cc
@@ -750,6 +750,14 @@ void NodeManager::ScheduleTasks() {
       EnqueuePlaceableTask(t);
     }
   }
+
+  // All remaining placeable tasks should be registered with the task dependency
+  // manager. TaskDependencyManager::TaskPending() is assumed to be idempotent.
+  // TODO(atumanov): evaluate performance implications of registering all new tasks on
+  // submission vs. registering remaining queued placeable tasks here.
+  for (const auto &task : local_queues_.GetPlaceableTasks()) {
+    task_dependency_manager_.TaskPending(task);
+  }
 }
 
 void NodeManager::TreatTaskAsFailed(const TaskSpecification &spec) {