Browse Source

Fix sc.

pull/1329/head
unknown 4 years ago
parent
commit
3123755f28
1 changed files with 3 additions and 4 deletions
  1. +3
    -4
      ge/graph/load/model_manager/davinci_model.cc

+ 3
- 4
ge/graph/load/model_manager/davinci_model.cc View File

@@ -2309,8 +2309,7 @@ Status DavinciModel::SinkTimeProfile(const InputData &current_data) {
}

Status DavinciModel::ProfileStepInfo(DavinciModel *model, uint16_t tag_id) {
auto &prof_mgr = ProfilingManager::Instance();
if (prof_mgr.ProfilingModelExecuteOn()) {
if (ProfilingManager::Instance().ProfilingModelExecuteOn()) {
uint64_t index_id = model->iterator_count_ + 1;
uint64_t model_id = static_cast<uint64_t>(model->Id());
rtError_t rt_ret = RT_ERROR_NONE;
@@ -2338,7 +2337,7 @@ Status DavinciModel::ProfileStepInfo(DavinciModel *model, uint16_t tag_id) {
return RT_ERROR_TO_GE_STATUS(rt_ret);
}
GELOGD("Get profiling args, task_id[%u], stream_id[%u]", task_id, stream_id);
uint32_t device_id = model->GetDeviceId();
int32_t device_id = static_cast<int32_t>(model->GetDeviceId());

Json step_info;
step_info[kIndexId] = index_id;
@@ -2359,7 +2358,7 @@ Status DavinciModel::ProfileStepInfo(DavinciModel *model, uint16_t tag_id) {
}
reported_data.append(",")
.append("\n");
prof_mgr.ReportData(device_id, reported_data, "step_info");
ProfilingManager::Instance().ReportData(device_id, reported_data, "step_info");
}
return SUCCESS;
}


Loading…
Cancel
Save