mirror of
https://github.com/wassname/ray.git
synced 2026-09-09 11:32:43 +08:00
catch exceptions on workers and pass them to the scheduler (#93)
This commit is contained in:
committed by
Philipp Moritz
parent
61a0014c00
commit
41539141af
+5
-3
@@ -263,15 +263,17 @@ Task* Worker::receive_next_task() {
|
||||
return task;
|
||||
}
|
||||
|
||||
void Worker::notify_task_completed() {
|
||||
void Worker::notify_task_completed(bool task_succeeded, std::string error_message) {
|
||||
if (!connected_) {
|
||||
RAY_LOG(RAY_FATAL, "Attempting to perform notify_task_completed, but connected_ = " << connected_ << ".");
|
||||
}
|
||||
ClientContext context;
|
||||
WorkerReadyRequest request;
|
||||
NotifyTaskCompletedRequest request;
|
||||
request.set_workerid(workerid_);
|
||||
request.set_task_succeeded(task_succeeded);
|
||||
request.set_error_message(error_message);
|
||||
AckReply reply;
|
||||
scheduler_stub_->WorkerReady(&context, request, &reply);
|
||||
scheduler_stub_->NotifyTaskCompleted(&context, request, &reply);
|
||||
}
|
||||
|
||||
void Worker::disconnect() {
|
||||
|
||||
Reference in New Issue
Block a user