Commit 11328f64 by ziheng Committed by Tianqi Chen

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

parent b40d43c4
......@@ -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();
}
}
......
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