Browse Source

!2416 Fix new control stream bug

Merge pull request !2416 from caifubi/fix-new-control-stream-bug
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
b7bc8e31f6
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