diff options
author | Guo, Yejun <yejun.guo@intel.com> | 2021-03-13 14:14:51 +0800 |
---|---|---|
committer | Guo, Yejun <yejun.guo@intel.com> | 2021-03-18 09:30:09 +0800 |
commit | da12d600ea6d16ea1b61d0b9a2530dea0bbbb870 (patch) | |
tree | b6734f5ea3e3794fb3622711390a9f88d15628a6 | |
parent | df59ae8bb21b596125eccaab7d7984d03dd9ab45 (diff) | |
download | ffmpeg-da12d600ea6d16ea1b61d0b9a2530dea0bbbb870.tar.gz |
lavfi/dnn_backend_openvino.c: fix mem leak for TaskItem upon error
-rw-r--r-- | libavfilter/dnn/dnn_backend_openvino.c | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/libavfilter/dnn/dnn_backend_openvino.c b/libavfilter/dnn/dnn_backend_openvino.c index 55cb0c757e..9a47d74c15 100644 --- a/libavfilter/dnn/dnn_backend_openvino.c +++ b/libavfilter/dnn/dnn_backend_openvino.c @@ -678,12 +678,6 @@ DNNReturnType ff_dnn_execute_model_async_ov(const DNNModel *model, const char *i return DNN_ERROR; } - task = av_malloc(sizeof(*task)); - if (!task) { - av_log(ctx, AV_LOG_ERROR, "unable to alloc memory for task item.\n"); - return DNN_ERROR; - } - if (!ov_model->exe_network) { if (init_model_ov(ov_model, input_name, output_names[0]) != DNN_SUCCESS) { av_log(ctx, AV_LOG_ERROR, "Failed init OpenVINO exectuable network or inference request\n"); @@ -691,6 +685,12 @@ DNNReturnType ff_dnn_execute_model_async_ov(const DNNModel *model, const char *i } } + task = av_malloc(sizeof(*task)); + if (!task) { + av_log(ctx, AV_LOG_ERROR, "unable to alloc memory for task item.\n"); + return DNN_ERROR; + } + task->done = 0; task->do_ioproc = 1; task->async = 1; |