add extensions for new Inference Engine for CPU.

Signed-off-by: Lewis Liu <[email protected]>
intel · Dec 4, 2019 · e7f0f36 · e7f0f36
1 parent bc1327c
commit e7f0f36
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dynamic_vino_lib/src/engines/engine_manager.cpp b/dynamic_vino_lib/src/engines/engine_manager.cpp
@@ -57,6 +57,9 @@ std::shared_ptr<Engines::Engine> Engines::EngineManager::createEngine_V2019R2_pl
   const std::string & device, const std::shared_ptr<Models::BaseModel> & model)
 {
   InferenceEngine::Core core;
+  if ((device.find("CPU") != std::string::npos)) {
+    core.AddExtension(std::make_shared<InferenceEngine::Extensions::Cpu::CpuExtensions>(), device);
+  }
   auto executable_network = core.LoadNetwork(model->getNetReader()->getNetwork(), device);
   auto request = executable_network.CreateInferRequestPtr();