issue #37

leafqycc · Apr 9, 2024 · d86b5a9 · d86b5a9
1 parent c385d9d
commit d86b5a9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/include/rknnPool.hpp b/include/rknnPool.hpp
@@ -80,6 +80,7 @@ int rknnPool<rknnModel, inputType, outputType>::getModelId()
 template <typename rknnModel, typename inputType, typename outputType>
 int rknnPool<rknnModel, inputType, outputType>::put(inputType inputData)
 {
+    std::lock_guard<std::mutex> lock(queueMtx);
     futs.push(pool->submit(&rknnModel::infer, models[this->getModelId()], inputData));
     return 0;
 }
@@ -105,4 +106,4 @@ rknnPool<rknnModel, inputType, outputType>::~rknnPool()
     }
 }
 
-#endif
+#endif