[Core] Task status should start with PENDING_ARGS_AVAIL when retry (r…

…ay-project#46494) Signed-off-by: Jiajun Yao <jeromeyjj@gmail.com>
icoco · Jul 9, 2024 · c9b14d7 · c9b14d7
1 parent 8f70e02
commit c9b14d7
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/ray/core_worker/task_manager.cc b/src/ray/core_worker/task_manager.cc
@@ -1395,9 +1395,10 @@ void TaskManager::MarkDependenciesResolved(const TaskID &task_id) {
   if (it == submissible_tasks_.end()) {
     return;
   }
-  if (it->second.GetStatus() == rpc::TaskStatus::PENDING_ARGS_AVAIL) {
-    SetTaskStatus(it->second, rpc::TaskStatus::PENDING_NODE_ASSIGNMENT);
-  }
+
+  RAY_CHECK(it->second.GetStatus() == rpc::TaskStatus::PENDING_ARGS_AVAIL)
+      << ", task ID = " << it->first << ", status = " << it->second.GetStatus();
+  SetTaskStatus(it->second, rpc::TaskStatus::PENDING_NODE_ASSIGNMENT);
 }
 
 void TaskManager::MarkTaskWaitingForExecution(const TaskID &task_id,
@@ -1458,12 +1459,12 @@ void TaskManager::MarkTaskRetryOnFailed(TaskEntry &task_entry,
   task_entry.MarkRetryOnFailed();
 
   // Mark the new status and also include task spec info for the new attempt.
-  task_entry.SetStatus(rpc::TaskStatus::PENDING_NODE_ASSIGNMENT);
+  task_entry.SetStatus(rpc::TaskStatus::PENDING_ARGS_AVAIL);
   RAY_UNUSED(RecordTaskStatusEventIfNeeded(task_entry.spec.TaskId(),
                                            task_entry.spec.JobId(),
                                            task_entry.spec.AttemptNumber() + 1,
                                            task_entry.spec,
-                                           rpc::TaskStatus::PENDING_NODE_ASSIGNMENT,
+                                           rpc::TaskStatus::PENDING_ARGS_AVAIL,
                                            /* include_task_info */ true));
 }