diff --git a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc index 07537067fd..c34c1b0713 100644 --- a/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc +++ b/mindspore/ccsrc/pipeline/pynative/pynative_execute.cc @@ -1810,8 +1810,17 @@ bool PynativeExecutor::ParseAssignExprNode(const std::shared_ptr(value_in_slice_node)) { + MS_LOG(DEBUG) << "Parse value node is none!"; + return false; + } const auto &node_name_in_slice_node = ParseNodeName(ast, value_in_slice_node, parse::AST_MAIN_TYPE_EXPR); - if (cell_input_args_.find(node_name_in_slice_node) != cell_input_args_.end()) { + std::string id; + if (py::hasattr(value_in_slice_node, "id")) { + id = py::cast(value_in_slice_node.attr("id")); + } + if (cell_input_args_.find(node_name_in_slice_node) != cell_input_args_.end() || + (!id.empty() && cell_input_args_.find(id) != cell_input_args_.end())) { return true; } }