From c0f5872e08bffa0369a812f28750df928fc6425d Mon Sep 17 00:00:00 2001 From: zhoufeng Date: Sat, 18 Jul 2020 18:07:55 +0800 Subject: [PATCH] skip case of not insert assign node Signed-off-by: zhoufeng --- mindspore/ccsrc/backend/session/ascend_control_parser.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mindspore/ccsrc/backend/session/ascend_control_parser.cc b/mindspore/ccsrc/backend/session/ascend_control_parser.cc index 4306a3d785..48a6a2325d 100644 --- a/mindspore/ccsrc/backend/session/ascend_control_parser.cc +++ b/mindspore/ccsrc/backend/session/ascend_control_parser.cc @@ -695,6 +695,9 @@ void AscendControlParser::InsertMultipleAssignToGraph(NotNull fr } for (size_t i = 0; i < from_outputs.size(); i++) { auto assign_node = InsertAssignToGraph(from_graph, NOT_NULL(from_outputs[i]), NOT_NULL(to_outputs[i])); + if (assign_node == nullptr) { + continue; + } const auto &from_graph_exe_order = from_graph->execution_order(); std::vector real_exe_order(from_graph_exe_order.size()); size_t real_exe_order_size = 0;