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

Visualize computation graph

Browse Source
This commit is contained in:
Philipp Moritz
2016-07-01 12:13:23 -07:00
parent ad35da08f3
commit 9ad2c033d4
7 changed files with 91 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/scheduler.cc
+8 -1
View File
@@ -49,7 +49,10 @@ Status SchedulerService::SubmitTask(ServerContext* context, const SubmitTaskRequ
Status SchedulerService::PutObj(ServerContext* context, const PutObjRequest* request, PutObjReply* reply) {
ObjRef objref = register_new_object();
ObjStoreId objstoreid = get_store(request->workerid());
auto operation = std::unique_ptr<Operation>(new Operation());
operation->mutable_put()->set_objref(objref);
operation->set_creator_operationid((*workers_.get())[request->workerid()].current_task);
computation_graph_.get()->add_operation(std::move(operation));
reply->set_objref(objref);
schedule();
return Status::OK;
@@ -59,6 +62,10 @@ Status SchedulerService::RequestObj(ServerContext* context, const RequestObjRequ
size_t size = objtable_.get()->size();
ObjRef objref = request->objref();
RAY_CHECK_LT(objref, size, "internal error: no object with objref " << objref << " exists");
auto operation = std::unique_ptr<Operation>(new Operation());
operation->mutable_get()->set_objref(objref);
operation->set_creator_operationid((*workers_.get())[request->workerid()].current_task);
computation_graph_.get()->add_operation(std::move(operation));
get_queue_.get()->push_back(std::make_pair(request->workerid(), objref));
schedule();
return Status::OK;