Browse Source

!13549 make IncorporateGetItemSwtich pass possible if there is env_setitem/env_getitem in that funcgraph

From: @xychow
Reviewed-by: @zh_qh,@ginfung
Signed-off-by: @zh_qh
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
aede003317
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/ccsrc/frontend/optimizer/irpass/incorporate_getitem.h

+ 2
- 1
mindspore/ccsrc/frontend/optimizer/irpass/incorporate_getitem.h View File

@@ -362,7 +362,8 @@ class IncorporateGetitemSwitch : public AnfVisitor {
is_in_get_ = false;

auto fg = node->func_graph();
if (idx_ == -1 || switch_ == nullptr || fg == nullptr || fg->has_flag(FUNC_GRAPH_FLAG_DEFER_INLINE)) {
if (idx_ == -1 || switch_ == nullptr || fg == nullptr ||
(fg->has_flag(FUNC_GRAPH_FLAG_DEFER_INLINE) && !ExistEnvNode(fg))) {
return nullptr;
}



Loading…
Cancel
Save