Skip to content

T
tic
Commit 11328f64 by ziheng Committed by Tianqi Chen
Options

[FIX] Pass the attributes of master node (#304)

parent b40d43c4
Showing with 2 additions and 2 deletions
apps/graph_executor/src/graph_pass.cc
...@@ -279,9 +279,9 @@ nnvm::Graph GraphFuse(nnvm::Graph g) { ...@@ -279,9 +279,9 @@ nnvm::Graph GraphFuse(nnvm::Graph g) {
int master = master_vec[root_id]; int master = master_vec[root_id];
CHECK_GE(master, 0); CHECK_GE(master, 0);
fe.schedule = fschedule[idx[master].source->op()]( fe.schedule = fschedule[idx[master].source->op()](
inode.source->attrs, fe.outputs, target); idx[master].source->attrs, fe.outputs, target);
std::ostringstream os; std::ostringstream os;
os << inode.source->attrs.name + "_id" << nid; os << idx[master].source->attrs.name + "_id" << nid;
fe.func_name = os.str(); fe.func_name = os.str();
} }
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment