wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-08-15 12:45:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

microbenchmarks and timing (#67)

Browse Source
This commit is contained in:
Robert Nishihara
2016-05-22 10:05:14 -07:00
committed by Philipp Moritz
parent f2824d134e
commit c4b783fbc8
4 changed files with 113 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/orchpylib.cc
+1 -3
View File
@@ -550,9 +550,7 @@ PyObject* deserialize_call(PyObject* self, PyObject* args) {
}
Worker* worker;
PyObjectToWorker(worker_capsule, &worker);
if (objrefs.size() > 0) {
worker->decrement_reference_count(objrefs);
}
worker->decrement_reference_count(objrefs);
int resultsize = call->result_size();
std::vector<ObjRef> result_objrefs;
PyObject* resultlist = PyList_New(resultsize);
src/worker.cc
+18 -14
View File
@@ -215,14 +215,16 @@ void Worker::increment_reference_count(std::vector<ObjRef> &objrefs) {
ORCH_LOG(ORCH_DEBUG, "Attempting to increment_reference_count for objrefs, but connected_ = " << connected_ << " so returning instead.");
return;
}
ClientContext context;
IncrementRefCountRequest request;
for (int i = 0; i < objrefs.size(); ++i) {
ORCH_LOG(ORCH_REFCOUNT, "Incrementing reference count for objref " << objrefs[i]);
request.add_objref(objrefs[i]);
if (objrefs.size() > 0) {
ClientContext context;
IncrementRefCountRequest request;
for (int i = 0; i < objrefs.size(); ++i) {
ORCH_LOG(ORCH_REFCOUNT, "Incrementing reference count for objref " << objrefs[i]);
request.add_objref(objrefs[i]);
}
AckReply reply;
scheduler_stub_->IncrementRefCount(&context, request, &reply);
}
AckReply reply;
scheduler_stub_->IncrementRefCount(&context, request, &reply);
}
void Worker::decrement_reference_count(std::vector<ObjRef> &objrefs) {
@@ -230,14 +232,16 @@ void Worker::decrement_reference_count(std::vector<ObjRef> &objrefs) {
ORCH_LOG(ORCH_DEBUG, "Attempting to decrement_reference_count, but connected_ = " << connected_ << " so returning instead.");
return;
}
ClientContext context;
DecrementRefCountRequest request;
for (int i = 0; i < objrefs.size(); ++i) {
ORCH_LOG(ORCH_REFCOUNT, "Decrementing reference count for objref " << objrefs[i]);
request.add_objref(objrefs[i]);
if (objrefs.size() > 0) {
ClientContext context;
DecrementRefCountRequest request;
for (int i = 0; i < objrefs.size(); ++i) {
ORCH_LOG(ORCH_REFCOUNT, "Decrementing reference count for objref " << objrefs[i]);
request.add_objref(objrefs[i]);
}
AckReply reply;
scheduler_stub_->DecrementRefCount(&context, request, &reply);
}
AckReply reply;
scheduler_stub_->DecrementRefCount(&context, request, &reply);
}
void Worker::register_function(const std::string& name, size_t num_return_vals) {