Browse Source

reorder for multi target run

tags/v1.1.0
kswang 5 years ago
parent
commit
c917fe4eb4
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/backend/session/ascend_session.cc

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

@@ -262,6 +262,9 @@ void AscendSession::BuildGraphImpl(GraphId graph_id) {
}
// adjust execution order because merge child graph and other special operations
AdjustKernel(graph);
// Reorder optimizer order
auto execution_order = graph->execution_order();
Reorder(&execution_order);
// Assign streams for control sink and hccl and so on
AssignStream(NOT_NULL(graph));



Loading…
Cancel
Save