diff --git a/mindspore/ccsrc/runtime/framework/actor/recorder_actor.cc b/mindspore/ccsrc/runtime/framework/actor/recorder_actor.cc index 3a027bad014..82723b8efd4 100644 --- a/mindspore/ccsrc/runtime/framework/actor/recorder_actor.cc +++ b/mindspore/ccsrc/runtime/framework/actor/recorder_actor.cc @@ -66,8 +66,9 @@ void RecorderActor::RecordOnStepEnd(OpContext *op_context) { MS_EXCEPTION_IF_NULL(op_context); // todo clear // Record iter_start, fp_start and iter_end op name and timestamp at the step end. (GPU) - profiler::ProfilerManager::GetInstance()->RecordOneStepStartEndInfo(); + if (profiler::ProfilerManager::GetInstance()->GetEnableRecorderActorFlag()) { + profiler::ProfilerManager::GetInstance()->RecordOneStepStartEndInfo(); + } } - } // namespace runtime } // namespace mindspore diff --git a/mindspore/ccsrc/runtime/framework/graph_scheduler.cc b/mindspore/ccsrc/runtime/framework/graph_scheduler.cc index 42e8d7c4241..2aade2fc62b 100644 --- a/mindspore/ccsrc/runtime/framework/graph_scheduler.cc +++ b/mindspore/ccsrc/runtime/framework/graph_scheduler.cc @@ -474,23 +474,12 @@ void GraphScheduler::BuildAndScheduleGlobalActor() { (void)actorMgr->Spawn(base_actor, false); // Create and schedule recorder actor. - bool recorder_actor_need = false; - if (profiler::ProfilerManager::GetInstance()->GetEnableRecorderActorFlag()) { - recorder_actor_need = true; - } -#ifdef ENABLE_DUMP_IR - if (mindspore::RecorderManager::Instance().RdrEnable()) { - recorder_actor_need = true; - } -#endif - if (recorder_actor_need) { - auto recorder_actor = std::make_shared(); - MS_EXCEPTION_IF_NULL(recorder_actor); - recorder_aid_ = &(recorder_actor->GetAID()); - auto base_recorder_actor = static_cast(recorder_actor); - base_recorder_actor->set_thread_pool(thread_pool_); - (void)actorMgr->Spawn(base_recorder_actor, true); - } + auto recorder_actor = std::make_shared(); + MS_EXCEPTION_IF_NULL(recorder_actor); + recorder_aid_ = &(recorder_actor->GetAID()); + auto base_recorder_actor = static_cast(recorder_actor); + base_recorder_actor->set_thread_pool(thread_pool_); + (void)actorMgr->Spawn(base_recorder_actor, true); // Create and schedule debug actor. bool debugger_actor_need = DumpJsonParser::GetInstance().e2e_dump_enabled();