IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[IE CLDNN] Fixed factory registration in concurrent environment (#3833)

Browse Source
This commit is contained in:
Vladimir Paramuzov 2021-01-14 09:36:14 +03:00 committed by GitHub
parent 9ddbfac6b1
commit 036259481d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
inference-engine/src/cldnn_engine/cldnn_program.h
View File

@ -9,6 +9,7 @@
#include <memory>
#include <string>
#include <cstdint>
#include <mutex>
#include <cpp/ie_cnn_network.h>
#include "details/ie_exception.hpp"
@ -126,7 +127,10 @@ public:
template<typename OpType, typename std::enable_if<std::is_base_of<ngraph::Node, OpType>::value, int>::type = 0>
static void RegisterFactory(factory_t func) {
Program::factories_map.insert({OpType::type_info, func});
static std::mutex m;
std::lock_guard<std::mutex> lock(m);
if (Program::factories_map.find(OpType::type_info) == Program::factories_map.end())
Program::factories_map.insert({OpType::type_info, func});
}
template<typename PType>