Skip to content

Commit

Permalink
Fixed segfault in ARM compute (#407)
Browse files Browse the repository at this point in the history
  • Loading branch information
ilyachur authored Aug 25, 2022
1 parent 25afef3 commit c0bac97
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions modules/arm_plugin/src/arm_executable_network.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -38,12 +38,12 @@ ArmPlugin::ExecutableNetwork::ExecutableNetwork(const std::shared_ptr<const ov::

void ArmPlugin::ExecutableNetwork::InitExecutor() {
if (_cfg._exclusiveAsyncRequests) {
_taskExecutor = ExecutorManager::getInstance()->getExecutor("CPU");
_taskExecutor = InferenceEngine::executorManager()->getExecutor("CPU");
} else {
auto streamsExecutorConfig = InferenceEngine::IStreamsExecutor::Config::MakeDefaultMultiThreaded(_cfg._streamsExecutorConfig);
streamsExecutorConfig._name = "CPUStreamsExecutor";
streamsExecutorConfig._threadBindingType = InferenceEngine::IStreamsExecutor::NONE;
_taskExecutor = ExecutorManager::getInstance()->getIdleCPUStreamsExecutor(streamsExecutorConfig);
_taskExecutor = InferenceEngine::executorManager()->getIdleCPUStreamsExecutor(streamsExecutorConfig);
}
_executor = _taskExecutor.get();
}
Expand Down
2 changes: 1 addition & 1 deletion modules/arm_plugin/src/arm_plugin.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ Plugin::~Plugin() {
std::lock_guard<std::mutex> lock{armSchedulerMutex};
arm_compute::Scheduler::set(arm_compute::Scheduler::Type::ST);
}
ExecutorManager::getInstance()->clear("CPUStreamsExecutor");
InferenceEngine::executorManager()->clear("CPUStreamsExecutor");
}

std::shared_ptr<ov::Model> Plugin::Transform(const std::shared_ptr<const ov::Model>& model,
Expand Down

0 comments on commit c0bac97

Please sign in to comment.