Browse Source

Pre Merge pull request !2050 from zhupuxu/delete_step

pull/2050/MERGE
zhupuxu Gitee 3 years ago
parent
commit
8a9544934c
2 changed files with 3 additions and 2 deletions
  1. +1
    -1
      ge/common/profiling/ge_profiling.cc
  2. +2
    -1
      ge/single_op/single_op.cc

+ 1
- 1
ge/common/profiling/ge_profiling.cc View File

@@ -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;


+ 2
- 1
ge/single_op/single_op.cc View File

@@ -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<TaskDescInfo> task_desc_info;


Loading…
Cancel
Save