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

fix bug of eliminate cast when trans through depend

This commit is contained in:
chenfei 2021-02-26 18:50:10 +08:00
parent 9ab234369b
commit 5d7a867118
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/frontend/optimizer/irpass/cast_eliminate.cc
View File

@ -60,6 +60,7 @@ AnfNodePtr CastSameTypeEliminater::operator()(const OptimizerPtr &, const AnfNod
}
if (src_type->type_id() == tgt_type->type_id()) {
// If 2nd input of cast is a depend, can't erase cast directly, but should replace cast with a new depend.
if (IsPrimitiveCNode(node->cast<CNodePtr>()->input(2), prim::kPrimDepend)) {
auto new_depend =
node->func_graph()->NewCNode({NewValueNode(prim::kPrimDepend), src_, node->cast<CNodePtr>()->input(2)});
@ -72,11 +73,10 @@ AnfNodePtr CastSameTypeEliminater::operator()(const OptimizerPtr &, const AnfNod
}
void CastSameTypeEliminater::Visit(const AnfNodePtr &node) {
auto cur_node = TransThroughDepend(node);
if (src_ == nullptr) {
src_ = cur_node;
src_ = node;
} else {
tgt_ = cur_node;
tgt_ = TransThroughDepend(node);
}
}