!13126 Set abstract value for reserved Depend node when merge Depend nodes.

From: @zh_qh
Reviewed-by: @ginfung
Signed-off-by: @ginfung
pull/13126/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit d556db2864

@ -579,7 +579,7 @@ void DumpIRProtoWithSrcInfo(const FuncGraphPtr &func_graph, const std::string &s
ChangeFileMode(file_path, S_IRUSR); ChangeFileMode(file_path, S_IRUSR);
} }
#else #else
void DumpIRProtoWithSrcInfo(const FuncGraphPtr &, const std::string &, const std::string &) { void DumpIRProtoWithSrcInfo(const FuncGraphPtr &, const std::string &, const std::string &, LocDebugDumpMode) {
static bool already_printed = false; static bool already_printed = false;
if (already_printed) { if (already_printed) {
return; return;

@ -327,6 +327,10 @@ void EliminateUselessNodesForUpdateStates(const std::vector<CNodePtr> &update_st
// If all users are Depend CNode. // If all users are Depend CNode.
if (depend_nodes.size() == us_users.size()) { if (depend_nodes.size() == us_users.size()) {
end = 1; end = 1;
// Set abstract value for reserved Depend node.
auto &reserved_depend_node = depend_nodes[0];
auto &primary_node = reserved_depend_node->cast<CNodePtr>()->input(kInputIndex);
reserved_depend_node->set_abstract(primary_node->abstract());
} }
for (ssize_t i = depend_nodes.size() - 1; i >= end; i--) { for (ssize_t i = depend_nodes.size() - 1; i >= end; i--) {
const auto &depend_node = depend_nodes[i]; const auto &depend_node = depend_nodes[i];

@ -382,7 +382,9 @@ void IrExportBuilder::SetShapeToNodeProto(const CNodePtr &node, mind_ir::NodePro
// 3. save tuple string in ref_attr_name // 3. save tuple string in ref_attr_name
MS_EXCEPTION_IF_NULL(node); MS_EXCEPTION_IF_NULL(node);
auto type = node->Type(); auto type = node->Type();
MS_EXCEPTION_IF_NULL(type);
auto shape = node->Shape(); auto shape = node->Shape();
MS_EXCEPTION_IF_NULL(shape);
ResetTupleIndex(); ResetTupleIndex();
std::string seq_string = "shape:"; std::string seq_string = "shape:";
mind_ir::AttributeProto *attr_proto = node_proto->add_attribute(); mind_ir::AttributeProto *attr_proto = node_proto->add_attribute();

Loading…
Cancel
Save