Browse Source

skip case of not insert assign node

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
tags/v0.6.0-beta
zhoufeng 5 years ago
parent
commit
c0f5872e08
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/backend/session/ascend_control_parser.cc

+ 3
- 0
mindspore/ccsrc/backend/session/ascend_control_parser.cc View File

@@ -695,6 +695,9 @@ void AscendControlParser::InsertMultipleAssignToGraph(NotNull<KernelGraphPtr> fr
} }
for (size_t i = 0; i < from_outputs.size(); i++) { 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])); 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(); const auto &from_graph_exe_order = from_graph->execution_order();
std::vector<CNodePtr> real_exe_order(from_graph_exe_order.size()); std::vector<CNodePtr> real_exe_order(from_graph_exe_order.size());
size_t real_exe_order_size = 0; size_t real_exe_order_size = 0;


Loading…
Cancel
Save