Browse Source

fix new control stream bug

tags/v0.5.0-beta
caifubi 5 years ago
parent
commit
0c29b70d06
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      mindspore/ccsrc/session/ascend_session.cc

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

@@ -282,6 +282,8 @@ GraphId AscendSession::CompileGraph(NotNull<FuncGraphPtr> func_graph) {
AdjustKernel(graph);
// assign stream
AssignStream(graph);
// insert profiling point
device::KernelAdjust::GetInstance().Profiling(NOT_NULL(graph.get()));
// build kernel
BuildKernel(graph);
// alloc mem


Loading…
Cancel
Save