virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-08-15 14:13:16 +02:00
Code Issues Releases Activity

dnn_backend_openvino.c: fix multi-thread issue for async execution

Browse Source 
once we mark done for the task in function infer_completion_callback,
the task is possible to be release in function ff_dnn_get_async_result_ov
in another thread just after it, so we need to record request queue
first, instead of using task->ov_model->request_queue later.

Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
This commit is contained in:
Guo, Yejun
2021-02-08 18:46:27 +08:00
parent 51c105a62d
commit 995c33a046
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
libavfilter/dnn/dnn_backend_openvino.c
View File

@@ -178,6 +178,7 @@ static void infer_completion_callback(void *args)
IEStatusCode status; IEStatusCode status;
RequestItem *request = args; RequestItem *request = args;
TaskItem *task = request->tasks[0]; TaskItem *task = request->tasks[0];
SafeQueue *requestq = task->ov_model->request_queue;
ie_blob_t *output_blob = NULL; ie_blob_t *output_blob = NULL;
ie_blob_buffer_t blob_buffer; ie_blob_buffer_t blob_buffer;
DNNData output; DNNData output;
@@ -243,7 +244,7 @@ static void infer_completion_callback(void *args)
request->task_count = 0; request->task_count = 0;
if (task->async) { if (task->async) {
if (ff_safe_queue_push_back(task->ov_model->request_queue, request) < 0) { if (ff_safe_queue_push_back(requestq, request) < 0) {
av_log(ctx, AV_LOG_ERROR, "Failed to push back request_queue.\n"); av_log(ctx, AV_LOG_ERROR, "Failed to push back request_queue.\n");
return; return;
} }