diff --git a/ge/common/profiling/ge_profiling.cc b/ge/common/profiling/ge_profiling.cc index a5857b35..cd51525a 100644 --- a/ge/common/profiling/ge_profiling.cc +++ b/ge/common/profiling/ge_profiling.cc @@ -268,7 +268,7 @@ ge::Status ProfSetStepInfo(uint64_t index_id, uint16_t tag_id, rtStream_t stream return ge::FAILED; } auto &profiling_manager = ge::ProfilingManager::Instance(); - profiling_manager.SetStepInfoIndex(index_id); + //profiling_manager.SetStepInfoIndex(index_id); if (is_first_run && tag_id == kStepStart) { GE_CHK_STATUS_RET_NOLOG(profiling_manager.ProfileStepInfo(index_id, kModelId, tag_id, stream, device_id)); is_first_run = false; diff --git a/ge/single_op/single_op.cc b/ge/single_op/single_op.cc index a82c30ba..73e41e8b 100755 --- a/ge/single_op/single_op.cc +++ b/ge/single_op/single_op.cc @@ -58,7 +58,8 @@ Status ProfilingTaskInfo(OpTask *op_task, const string &shape_type) { tmp_task_desc_info.op_name.c_str(), tmp_task_desc_info.model_name.c_str()); tmp_task_desc_info.shape_type = shape_type; - tmp_task_desc_info.cur_iter_num = ProfilingManager::Instance().GetStepInfoIndex(); + tmp_task_desc_info.cur_iter_num = 0; + //ProfilingManager::Instance().GetStepInfoIndex(); tmp_task_desc_info.task_type = op_task->GetTaskType(); std::vector task_desc_info;