From 1b3de31ff192f411c2d7aea7f816ac798e683832 Mon Sep 17 00:00:00 2001 From: Stephanie Wang Date: Thu, 23 Aug 2018 14:44:47 -0700 Subject: [PATCH] [xray] Fix bug where driver task ID is assumed to be nil (#2725) ## What do these changes do? #2362 left a bug where it assumed that the driver task ID was nil. This fixes the bug to check the `SchedulingQueue` for any driver task IDs instead. --- src/ray/raylet/node_manager.cc | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/ray/raylet/node_manager.cc b/src/ray/raylet/node_manager.cc index 641db346c..1d5d3da2f 100644 --- a/src/ray/raylet/node_manager.cc +++ b/src/ray/raylet/node_manager.cc @@ -1278,10 +1278,8 @@ void NodeManager::HandleObjectMissing(const ObjectID &object_id) { // Check that remaining tasks that could not be transitioned are running // workers or drivers, now blocked in a get. local_queues_.FilterState(waiting_task_id_set, TaskState::RUNNING); - if (!waiting_task_id_set.empty()) { - RAY_CHECK(waiting_task_id_set.size() == 1); - RAY_CHECK(waiting_task_id_set.begin()->is_nil()); - } + local_queues_.FilterState(waiting_task_id_set, TaskState::DRIVER); + RAY_CHECK(waiting_task_id_set.empty()); } }