[Python API] Fix benchmark hanging (#10457)
This commit is contained in:
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
|
||||
|
||||
Reference in new issue
Block a user