Add lock_guard in callbackSync::callback to prevent deadlock

This patch fixes a deadlock issue of NPU_INFER_BLOCKING mode.

Signed-off-by: Dongjin Na <dongjin.na@samsung.com>

diff --git a/src/core/ne-handler.cc b/src/core/ne-handler.cc
index 0516404edb3c93f83f9fbff83a038a6e54fa1909..34eca91f5fb76da413339852005b813510104fcb 100644 (file)
--- a/src/core/ne-handler.cc
+++ b/src/core/ne-handler.cc
@@ -302,7 +302,10 @@ class callbackSync {
   }
 
   void callback (output_buffers *output, int req_id) {
-    done_ = true;
+    {
+      std::lock_guard<std::mutex> lock (m_);
+      done_ = true;
+    }
     cv_.notify_one ();
   }