wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-02 05:16:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Set the workerid in submit_task and get_objref

Browse Source
This commit is contained in:
Philipp Moritz
2016-06-28 21:41:31 -07:00
parent 15a34fe9bf
commit 891dccb0ff
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/worker.cc
+2
View File
@@ -30,6 +30,7 @@ SubmitTaskReply Worker::submit_task(SubmitTaskRequest* request, int max_retries,
RAY_CHECK(connected_, "Attempted to perform submit_task but failed.");
SubmitTaskReply reply;
Status status;
request->set_workerid(workerid_);
for (int i = 0; i < 1 + max_retries; ++i) {
ClientContext context;
status = scheduler_stub_->SubmitTask(&context, *request, &reply);
@@ -73,6 +74,7 @@ ObjRef Worker::get_objref() {
// first get objref for the new object
RAY_CHECK(connected_, "Attempted to perform get_objref but failed.");
PutObjRequest request;
request.set_workerid(workerid_);
PutObjReply reply;
ClientContext context;
Status status = scheduler_stub_->PutObj(&context, request, &reply);