Browse Source

Feature: handle model_exit bug

tags/v1.1.0
l00444296 3 years ago
parent
commit
d6c515e572
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/graph/load/new_model_manager/davinci_model.cc

+ 2
- 2
ge/graph/load/new_model_manager/davinci_model.cc View File

@@ -2805,7 +2805,7 @@ void *DavinciModel::Run(DavinciModel *model) {
GELOGI("rtStreamSynchronize start."); GELOGI("rtStreamSynchronize start.");
rt_ret = rtStreamSynchronize(model->rt_model_stream_); rt_ret = rtStreamSynchronize(model->rt_model_stream_);
if (rt_ret == RT_ERROR_MODEL_ABORT_NORMAL) { if (rt_ret == RT_ERROR_MODEL_ABORT_NORMAL) {
GELOGI("Stream Synchronize get result to abort model normal");
GELOGI("The model with multiple datasets aborts normally.");
} else { } else {
GE_IF_BOOL_EXEC(rt_ret != RT_ERROR_NONE, rslt_flg = false; GE_IF_BOOL_EXEC(rt_ret != RT_ERROR_NONE, rslt_flg = false;
(void)model->ReturnResult(current_data.index, false, seq_end_flag, data_wrapper->GetOutput()); (void)model->ReturnResult(current_data.index, false, seq_end_flag, data_wrapper->GetOutput());
@@ -2833,7 +2833,7 @@ void *DavinciModel::Run(DavinciModel *model) {
seq_end_flag = true; seq_end_flag = true;
} }
if (rt_ret == RT_ERROR_MODEL_ABORT_NORMAL) { if (rt_ret == RT_ERROR_MODEL_ABORT_NORMAL) {
GELOGI("Stream Synchronize get result to abort model normal");
GELOGI("The model with multiple datasets aborts normally.");
} else { } else {
GE_IF_BOOL_EXEC( GE_IF_BOOL_EXEC(
rt_ret != RT_ERROR_NONE, rslt_flg = false; GELOGI("seq_end_flg: %d", seq_end_flag); rt_ret != RT_ERROR_NONE, rslt_flg = false; GELOGI("seq_end_flg: %d", seq_end_flag);


Loading…
Cancel
Save