diff options
Diffstat (limited to 'src/backends/backendsCommon/DynamicBackend.cpp')
-rw-r--r-- | src/backends/backendsCommon/DynamicBackend.cpp | 34 |
1 files changed, 27 insertions, 7 deletions
diff --git a/src/backends/backendsCommon/DynamicBackend.cpp b/src/backends/backendsCommon/DynamicBackend.cpp index 06d819b9a7..c576199e1f 100644 --- a/src/backends/backendsCommon/DynamicBackend.cpp +++ b/src/backends/backendsCommon/DynamicBackend.cpp @@ -70,18 +70,22 @@ BackendVersion DynamicBackend::GetBackendVersion() IBackendInternalUniquePtr DynamicBackend::GetBackend() { + // This call throws in case of error + return CreateBackend(); +} + +BackendRegistry::FactoryFunction DynamicBackend::GetFactoryFunction() +{ if (m_BackendFactoryFunction == nullptr) { - throw RuntimeException("GetBackend error: invalid function pointer"); + throw RuntimeException("GetFactoryFunction error: invalid function pointer"); } - auto backendPointer = reinterpret_cast<IBackendInternal*>(m_BackendFactoryFunction()); - if (backendPointer == nullptr) + return [this]() -> IBackendInternalUniquePtr { - throw RuntimeException("GetBackend error: backend instance must not be null"); - } - - return std::unique_ptr<IBackendInternal>(backendPointer); + // This call throws in case of error + return CreateBackend(); + }; } template<typename BackendFunctionType> @@ -108,4 +112,20 @@ BackendFunctionType DynamicBackend::SetFunctionPointer(const std::string& backen return functionPointer; } +IBackendInternalUniquePtr DynamicBackend::CreateBackend() +{ + if (m_BackendFactoryFunction == nullptr) + { + throw RuntimeException("CreateBackend error: invalid function pointer"); + } + + auto backendPointer = reinterpret_cast<IBackendInternal*>(m_BackendFactoryFunction()); + if (backendPointer == nullptr) + { + throw RuntimeException("CreateBackend error: backend instance must not be null"); + } + + return std::unique_ptr<IBackendInternal>(backendPointer); +} + } // namespace armnn |