modified: tests/ut/ge/graph/passes/base_pass_unittest.cc

4 years ago · 2572bed425
parent f203c70cfd
commit 2572bed425
1 changed files with 1 additions and 0 deletions
--- a/tests/ut/ge/graph/passes/base_pass_unittest.cc
+++ b/tests/ut/ge/graph/passes/base_pass_unittest.cc
@ -75,6 +75,7 @@ class UtestTestPass : public BaseNodePass {
        return SUCCESS;
      }
      if(need_repass){
+        AttrUtils::SetBool(node->GetOpDesc(),"need_infer_again_", false);
        AddImmediateRePassNode(node);
      }
      else{