[Python API] Fix benchmark hanging (#10457)

This commit is contained in:
Anastasia Kuporosova authored and GitHub committed 2022-02-17 10:59:55 +03:00
1 parent be6a3c34f1
commit 861d43e06d
1 file changed
+4 -1
@@ -183,7 +183,10 @@ void regclass_AsyncInferQueue(py::module m) {
// getIdleRequestId function has an intention to block InferQueue
// until there is at least one idle (free to use) InferRequest
auto handle = self.get_idle_request_id();
self._idle_handles.pop();
{
std::lock_guard<std::mutex> lock(self._mutex);
self._idle_handles.pop();
}
// Set new inputs label/id from user
self._user_ids[handle] = userdata;
// Update inputs if there are any