Browse Source

Fix bug.

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

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

@@ -3754,18 +3754,18 @@ Status DavinciModel::NnExecute(rtStream_t stream, bool async_mode, const InputDa
GELOGD("current_data.index=%u", input_data.index);
GE_IF_BOOL_EXEC(ProfilingManager::Instance().ProfilingModelExecuteOn(), SetProfileTime(MODEL_PRE_PROC_END));

// tag_id 0 means step begin, 1 meas step end.
GE_CHK_STATUS_RET_NOLOG(ProfileStepInfo(this, 0));
if (!task_list_.empty()) {
// tag_id 0 means step begin, 1 meas step end.
GE_CHK_STATUS_RET_NOLOG(ProfileStepInfo(this, 0));
GELOGD("rtModelExecute do");
GE_IF_BOOL_EXEC(ProfilingManager::Instance().ProfilingModelExecuteOn(), SetProfileTime(MODEL_INFER_START));
rtError_t rt_ret = rtModelExecute(rt_model_handle_, rt_model_stream_, 0);
GE_CHK_RT_EXEC(rt_ret, return RT_ERROR_TO_GE_STATUS(rt_ret));
GE_IF_BOOL_EXEC(ProfilingManager::Instance().ProfilingModelExecuteOn(), SetProfileTime(MODEL_INFER_END));
GELOGD("rtModelExecute end");
GE_CHK_STATUS_RET_NOLOG(ProfileStepInfo(this, 1));
iterator_count_++;
}
GE_CHK_STATUS_RET_NOLOG(ProfileStepInfo(this, 1));

if (!is_async_mode_) {
GE_IF_BOOL_EXEC(ProfilingManager::Instance().ProfilingModelExecuteOn(), SetProfileTime(MODEL_AFTER_PROC_START));


Loading…
Cancel
Save