MrBesen/ffmpeg
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

lavfi/dnn_backend_openvino.c: Fix Memory Leak in execute_model_ov

Browse Source 
In cases where the execution inside the function execute_model_ov fails,
the OVRequestItem must be pushed back to the request_queue before returning
the error. In case pushing back fails, release the allocated memory.

Signed-off-by: Shubhanshu Saxena <shubhanshu.e01@gmail.com>
This commit is contained in:
Shubhanshu Saxena 2021-07-11 21:45:01 +05:30 committed by Guo Yejun
parent c866a099b2
commit 429954822c
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
libavfilter/dnn/dnn_backend_openvino.c
View File

@ -432,6 +432,8 @@ static DNNReturnType execute_model_ov(RequestItem *request, Queue *inferenceq)
OVModel *ov_model;
if (ff_queue_size(inferenceq) == 0) {
ie_infer_request_free(&request->infer_request);
av_freep(&request);
return DNN_SUCCESS;
}
@ -443,7 +445,7 @@ static DNNReturnType execute_model_ov(RequestItem *request, Queue *inferenceq)
if (task->async) {
ret = fill_model_input_ov(ov_model, request);
if (ret != DNN_SUCCESS) {
return ret;
goto err;
}
status = ie_infer_set_completion_callback(request->infer_request, &request->callback);
if (status != OK) {
@ -459,7 +461,7 @@ static DNNReturnType execute_model_ov(RequestItem *request, Queue *inferenceq)
} else {
ret = fill_model_input_ov(ov_model, request);
if (ret != DNN_SUCCESS) {
return ret;
goto err;
}
status = ie_infer_request_infer(request->infer_request);
if (status != OK) {