From 41229ed01dad1b99a27f55ee96430dffbd38ca66 Mon Sep 17 00:00:00 2001 From: wuyongkang Date: Thu, 9 Jul 2020 14:51:33 +0800 Subject: [PATCH] Fix bug of for i, j in enumerate(items) --- mindspore/ccsrc/pipeline/parse/parse.cc | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/mindspore/ccsrc/pipeline/parse/parse.cc b/mindspore/ccsrc/pipeline/parse/parse.cc index 351a83124e..1d306d9ca4 100644 --- a/mindspore/ccsrc/pipeline/parse/parse.cc +++ b/mindspore/ccsrc/pipeline/parse/parse.cc @@ -1152,7 +1152,6 @@ FunctionBlockPtr Parser::ParseForLoop(const FunctionBlockPtr &block, const py::o // get varibale name of 'x' in statement 'for x in xs' py::object target_node = python_adapter::GetPyObjAttr(node, "target"); - auto name_id = py::cast(python_adapter::GetPyObjAttr(target_node, "id")); // create statement 'len(xs)' py::object iter_obj = python_adapter::GetPyObjAttr(node, "iter"); @@ -1174,13 +1173,11 @@ FunctionBlockPtr Parser::ParseForLoop(const FunctionBlockPtr &block, const py::o body_block->AddPrevBlock(header_block); // create 'x = xs[i]' CNodePtr target_var = body_block->func_graph()->NewCNode({op_getitem, iter_node, loop_var}); - target_var->debug_info()->set_name(name_id); - body_block->WriteVariable(name_id, target_var); + WriteAssignVars(body_block, target_node, target_var); // create 'i = i + 1' CNodePtr loop_var_inc = body_block->func_graph()->NewCNode({NewValueNode(prim::kPrimScalarAdd), loop_var, NewValueNode(1)}); body_block->WriteVariable(loop_var->name(), loop_var_inc); - loop_var_inc->debug_info()->set_name(name_id); // link the variable name with the target auto it_info = std::make_shared(loop_var_inc->debug_info());