p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!22439 mindir: Compatible CNode Name 

Merge pull request !22439 from lanzhineng/mindir_control_flow
This commit is contained in:
i-robot 2021-08-27 13:16:56 +00:00 committed by Gitee
parent 1639023909 a255e2b7c9
commit a96ca1b28f
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/transform/express_ir/mindir_exporter.cc
View File

@ -553,9 +553,13 @@ std::string IrExportBuilder::GetUniqueNodeName(const AnfNodePtr &node) {
return iter->second;
} else {
std::string node_name = GetNodeName(node);
// Compatible before. CNode = FuncGraphName:CNodeName:index ,Parameter = FuncGraphName:ParameterName
if (node->isa<CNode>()) {
node_name = node_name + ":" + std::to_string(GetNodeIndex());
}
// Avoid duplicate name.
while (nodeName_.count(node_name) > 0) {
auto node_idx = GetNodeIndex();
node_name = node_name + ":" + std::to_string(node_idx);
node_name = node_name + "_" + std::to_string(GetNodeIndex());
}
node_index_map_[node] = node_name;
return node_name;