fix new control stream bug

5 years ago · 0c29b70d06
parent 78a8bc302d
commit 0c29b70d06
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