From 891dccb0ff7b3f4bfc0f487491c7b65c601df74f Mon Sep 17 00:00:00 2001
From: Philipp Moritz <pcmoritz@gmail.com>
Date: Tue, 28 Jun 2016 21:41:31 -0700
Subject: [PATCH] Set the workerid in submit_task and get_objref

---
 src/worker.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/worker.cc b/src/worker.cc
index 950e7d3e1..d79060430 100644
--- a/src/worker.cc
+++ b/src/worker.cc
@@ -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);