Skip to content
Snippets Groups Projects
Commit 11328f64 authored by ziheng's avatar ziheng Committed by Tianqi Chen
Browse files

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

parent b40d43c4
No related branches found
No related tags found
No related merge requests found
......@@ -279,9 +279,9 @@ nnvm::Graph GraphFuse(nnvm::Graph g) {
int master = master_vec[root_id];
CHECK_GE(master, 0);
fe.schedule = fschedule[idx[master].source->op()](
inode.source->attrs, fe.outputs, target);
idx[master].source->attrs, fe.outputs, target);
std::ostringstream os;
os << inode.source->attrs.name + "_id" << nid;
os << idx[master].source->attrs.name + "_id" << nid;
fe.func_name = os.str();
}
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment