Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!19870 Bugfix for gpu profiler in callback mode missing step trace data 

Merge pull request !19870 from gzhcv/FitLoopCount
This commit is contained in:
i-robot 2021-07-10 07:32:56 +00:00 committed by Gitee
parent f10786f063 9d4258b95b
commit 722428d2ab
2 changed files with 9 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/runtime/framework/actor/recorder_actor.cc
View File

@ -66,8 +66,9 @@ void RecorderActor::RecordOnStepEnd(OpContext<DeviceTensor> *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

23
mindspore/ccsrc/runtime/framework/graph_scheduler.cc
View File

@ -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<RecorderActor>();
MS_EXCEPTION_IF_NULL(recorder_actor);
recorder_aid_ = &(recorder_actor->GetAID());
auto base_recorder_actor = static_cast<ActorReference>(recorder_actor);
base_recorder_actor->set_thread_pool(thread_pool_);
(void)actorMgr->Spawn(base_recorder_actor, true);
}
auto recorder_actor = std::make_shared<RecorderActor>();
MS_EXCEPTION_IF_NULL(recorder_actor);
recorder_aid_ = &(recorder_actor->GetAID());
auto base_recorder_actor = static_cast<ActorReference>(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();