!2416 Fix new control stream bug

Merge pull request !2416 from caifubi/fix-new-control-stream-bug
5 years ago · b7bc8e31f6
parent 886dfe6fd7 0c29b70d06
commit b7bc8e31f6
1 changed files with 2 additions and 0 deletions
--- a/mindspore/ccsrc/session/ascend_session.cc
+++ b/mindspore/ccsrc/session/ascend_session.cc
@ -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