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

Write computation graph to file

Browse Source
This commit is contained in:
Philipp Moritz
2016-06-27 12:20:30 -07:00
parent fb6c74ecab
commit 7af0f1b221
13 changed files with 96 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/computation_graph.cc
+6
View File
@@ -19,3 +19,9 @@ const Task& ComputationGraph::get_task(OperationId operationid) {
RAY_CHECK(operations_[operationid]->has_task(), "Calling get_task with operationid " << operationid << ", but this corresponds to a put not a task.");
return operations_[operationid]->task();
}
void ComputationGraph::to_protobuf(CompGraph* computation_graph) {
for (OperationId id = 0; id < operations_.size(); ++id) {
computation_graph->add_operation()->CopyFrom(*operations_[id]);
}
}