!14244 optimize exception mode when use undefined name in if for and while

From: @zhangbuxue
Reviewed-by: @ginfung,@zh_qh
Signed-off-by: @zh_qh

mindspore/ccsrc/pipeline/jit/parse/function_block.cc

@@ -169,6 +169,9 @@ AnfNodePtr FunctionBlock::MakeResolveSymbol(const std::string &value) {
     return MakeResolveClassMember(bits_str);
   }
   py::tuple namespace_var = parser_.ast()->CallParserObjMethod(PYTHON_PARSE_GET_NAMESPACE_SYMBOL, value);
+  if (namespace_var[0].is_none()) {
+    MS_EXCEPTION(NameError) << "The name \'" << value << "\' is not defined.";
+  }
 
   NameSpacePtr name_space = std::make_shared<NameSpace>(RESOLVE_NAMESPACE_NAME_SYMBOL_STR, namespace_var[0]);
   SymbolPtr symbol = std::make_shared<Symbol>(namespace_var[1].cast<std::string>());
@@ -193,6 +196,7 @@ AnfNodePtr FunctionBlock::MakeResolve(const NameSpacePtr &name_space, const Symb
 
 // Add input for the block's phi parameter
 void FunctionBlock::SetPhiArgument(const ParameterPtr &phi) {
+  TraceGuard trace_guard(std::make_shared<TraceResolve>(phi->debug_info()));
   std::string var = phi_nodes_[phi];
   MS_LOG(DEBUG) << "graph " << func_graph_->ToString() << " set phi " << phi->ToString() << " for var " << var;
   auto removable = CollectRemovablePhi(phi);
@@ -205,7 +209,6 @@ void FunctionBlock::SetPhiArgument(const ParameterPtr &phi) {
     MS_EXCEPTION_IF_NULL(pred);
     MS_LOG(DEBUG) << "graph " << func_graph_->ToString() << " pred_blocks_ " << pred->func_graph_->ToString();
     AnfNodePtr arg_node = pred->ReadVariable(var);
-    arg_node->set_debug_info(phi->debug_info());
     CNodePtr jump = pred->jumps_[this];
     jump->add_input(arg_node);
   }
@@ -264,7 +267,6 @@ bool FunctionBlock::CollectRemovablePhi(const ParameterPtr &phi) {
   }
   AnfNodePtr arg_node = SearchReplaceNode(var, phi);
   if (arg_node != nullptr) {
-    arg_node->set_debug_info(phi->debug_info());
     MS_LOG(DEBUG) << "graph " << func_graph_->ToString() << " phi " << phi->ToString() << " can be replaced with "
                   << arg_node->DebugString();
     // Replace var with new one. This equal to statement in TR "v0 is immediately replaced by v1."
@@ -417,7 +419,7 @@ void FunctionBlock::AttachIsolatedNodesBeforeReturn() {
   states.emplace_back(NewValueNode(prim::kPrimMakeTuple));
   for (auto &node : isolated_nodes_) {
     MS_LOG(DEBUG) << "Adding dependency, node: " << node->DebugString(2) << " in " << func_graph()->ToString();
-    if (node->func_graph() == func_graph() || node->isa<Parameter>()) {
+    if (node->func_graph() == func_graph()) {
       states.emplace_back(node);
     } else {
       MS_LOG(INFO) << "Ignored FV dependency, node: " << node->DebugString(2) << " in " << func_graph()->ToString();

tests/ut/cpp/pipeline/parse/parser_test.cc

@@ -380,5 +380,12 @@ TEST_F(TestParser, TestParseGraphCallVargs) {
   bool ret_ = ResolveAll(manager);
   ASSERT_TRUE(ret_);
 }
+
+TEST_F(TestParser, TestParserUndefinedVar) {
+  py::function fn_ = python_adapter::GetPyFn("gtest_input.pipeline.parse.parser_test", "test_parse_undefined_var");
+
+  // parse undefined var
+  EXPECT_THROW({ ParsePythonCode(fn_); }, std::runtime_error);
+}
 }  // namespace parse
 }  // namespace mindspore

tests/ut/cpp/pipeline/parse/resolve_test.cc

@@ -86,18 +86,5 @@ TEST_F(TestResolve, TestParseGraphTestClosureResolve) {
     i++;
   }
 }
-TEST_F(TestResolve, TestResolveFail) {
-  py::function fn_ = python_adapter::GetPyFn("gtest_input.pipeline.parse.parser_test", "test_resolvefail");
-
-  // parse graph
-  FuncGraphPtr func_graph = ParsePythonCode(fn_);
-  ASSERT_FALSE(nullptr == func_graph);
-
-  // save the func_graph to manager
-  std::shared_ptr<FuncGraphManager> manager = Manage(func_graph);
-
-  // call resolve
-  EXPECT_THROW({ ResolveAll(manager); }, std::runtime_error);
-}
 }  // namespace parse
 }  // namespace mindspore

tests/ut/cpp/python_input/gtest_input/pipeline/parse/parser_test.py

@@ -298,7 +298,7 @@ def test_augassign(x, y):
     return y
 
 
-def test_resolvefail(x, y):
+def test_parse_undefined_var(x, y):
     a = x + y + Undef
     return a