Browse Source

lavfi/dnn_backend_openvino.c: fix mem leak for RequestItem upon error

tags/n4.4
Guo, Yejun 4 years ago
parent
commit
df59ae8bb2
1 changed files with 7 additions and 9 deletions
  1. +7
    -9
      libavfilter/dnn/dnn_backend_openvino.c

+ 7
- 9
libavfilter/dnn/dnn_backend_openvino.c View File

@@ -353,25 +353,23 @@ static DNNReturnType init_model_ov(OVModel *ov_model, const char *input_name, co
goto err;
}

item->callback.completeCallBackFunc = infer_completion_callback;
item->callback.args = item;
if (ff_safe_queue_push_back(ov_model->request_queue, item) < 0) {
av_freep(&item);
goto err;
}

status = ie_exec_network_create_infer_request(ov_model->exe_network, &item->infer_request);
if (status != OK) {
av_freep(&item);
goto err;
}

item->tasks = av_malloc_array(ctx->options.batch_size, sizeof(*item->tasks));
if (!item->tasks) {
av_freep(&item);
goto err;
}
item->task_count = 0;

item->callback.completeCallBackFunc = infer_completion_callback;
item->callback.args = item;
if (ff_safe_queue_push_back(ov_model->request_queue, item) < 0) {
av_freep(&item);
goto err;
}
}

ov_model->task_queue = ff_queue_create();


Loading…
Cancel
Save