mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

do not throw exception when the forward netword is called more than once

This commit is contained in:
huangbingjian 2021-04-09 16:04:51 +08:00
parent ef27f1f663
commit 021c50cbc5
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/frontend/optimizer/ad/dfunctor.cc
View File

@ -825,8 +825,9 @@ static std::vector<std::pair<CNodePtr, CNodePtr>> FindPrimalJPair(const FuncGrap
auto fg = cnode->func_graph();
MS_EXCEPTION_IF_NULL(fg);
if (primal_users_map.find(fg) != primal_users_map.end()) {
MS_LOG(EXCEPTION) << "The forward network is only allowed to be called once. Func graph: " << fg->ToString()
<< ", cnode: " << cnode->DebugString() << ", trace: " << trace::DumpSourceLines(cnode);
MS_LOG(WARNING) << "It is recommended to call the forward network only once. Func graph: " << fg->ToString()
<< ", cnode: " << cnode->DebugString() << ", trace: " << trace::DumpSourceLines(cnode);
continue;
}
primal_users_map[fg] = cnode;
} else if (IsPrimitive(cnode->inputs().at(0), prim::kPrimJ)) {