From 0c29b70d06e5e4f16968586a5ded14a7465b16b5 Mon Sep 17 00:00:00 2001 From: caifubi Date: Sat, 20 Jun 2020 19:18:05 +0800 Subject: [PATCH] fix new control stream bug --- mindspore/ccsrc/session/ascend_session.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/mindspore/ccsrc/session/ascend_session.cc b/mindspore/ccsrc/session/ascend_session.cc index 7173a26ed1..f97a4d1e98 100644 --- a/mindspore/ccsrc/session/ascend_session.cc +++ b/mindspore/ccsrc/session/ascend_session.cc @@ -282,6 +282,8 @@ GraphId AscendSession::CompileGraph(NotNull 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