summaryrefslogtreecommitdiff
path: root/libavfilter/dnn
diff options
context:
space:
mode:
authorGuo, Yejun <yejun.guo@intel.com>2021-03-13 14:14:51 +0800
committerGuo, Yejun <yejun.guo@intel.com>2021-03-18 09:30:09 +0800
commitda12d600ea6d16ea1b61d0b9a2530dea0bbbb870 (patch)
treeb6734f5ea3e3794fb3622711390a9f88d15628a6 /libavfilter/dnn
parentdf59ae8bb21b596125eccaab7d7984d03dd9ab45 (diff)
lavfi/dnn_backend_openvino.c: fix mem leak for TaskItem upon error
Diffstat (limited to 'libavfilter/dnn')
-rw-r--r--libavfilter/dnn/dnn_backend_openvino.c12
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;