|
@@ -3701,12 +3701,14 @@ Status DavinciModel::NnExecute(rtStream_t stream, bool async_mode, const InputDa |
|
|
GE_CHK_STATUS_RET_NOLOG( |
|
|
GE_CHK_STATUS_RET_NOLOG( |
|
|
ProfilingManager::Instance().ProfileStepInfo(index_id, model_id, 0, rt_model_stream_, device_id)); |
|
|
ProfilingManager::Instance().ProfileStepInfo(index_id, model_id, 0, rt_model_stream_, device_id)); |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
GELOGD("rtModelExecute do"); |
|
|
GELOGD("rtModelExecute do"); |
|
|
GE_IF_BOOL_EXEC(profiling_model_execute_on, SetProfileTime(MODEL_INFER_START)); |
|
|
GE_IF_BOOL_EXEC(profiling_model_execute_on, SetProfileTime(MODEL_INFER_START)); |
|
|
rtError_t rt_ret = rtModelExecute(rt_model_handle_, rt_model_stream_, 0); |
|
|
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_CHK_RT_EXEC(rt_ret, return RT_ERROR_TO_GE_STATUS(rt_ret)); |
|
|
GE_IF_BOOL_EXEC(profiling_model_execute_on, SetProfileTime(MODEL_INFER_END)); |
|
|
GE_IF_BOOL_EXEC(profiling_model_execute_on, SetProfileTime(MODEL_INFER_END)); |
|
|
GELOGD("rtModelExecute end"); |
|
|
GELOGD("rtModelExecute end"); |
|
|
|
|
|
|
|
|
if (profiling_model_execute_on) { |
|
|
if (profiling_model_execute_on) { |
|
|
GE_CHK_STATUS_RET_NOLOG( |
|
|
GE_CHK_STATUS_RET_NOLOG( |
|
|
ProfilingManager::Instance().ProfileStepInfo(index_id, model_id, 1, rt_model_stream_, device_id)); |
|
|
ProfilingManager::Instance().ProfileStepInfo(index_id, model_id, 1, rt_model_stream_, device_id)); |
|
|