aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/DynamicBackendUtils.cpp
diff options
context:
space:
mode:
authorMatteo Martincigh <matteo.martincigh@arm.com>2019-08-05 14:12:11 +0100
committerMatteo Martincigh <matteo.martincigh@arm.com>2019-08-09 08:49:22 +0100
commit0c2b2897cb65b562b97014ae748e91d8b9a3d3b4 (patch)
treef4833fedc412ab4c434ca3c4181389f54b6487d2 /src/backends/backendsCommon/DynamicBackendUtils.cpp
parente54aa06ec19813b737513ecb8224285d98e871ba (diff)
downloadarmnn-0c2b2897cb65b562b97014ae748e91d8b9a3d3b4.tar.gz
IVGCVSW-3596 Register the dynamic backends in the BackendRegistry
* Added getter for the factory function in the DynamicBackend class * Added new RegisterDynamicBackends method in utils class * Added dynamic backend registration process in the Runtime class * Added new dummy dynamic backend objects for testing * Added unit tests for dynamic backend registration * Added convenience methods to BackendId Change-Id: I01e147d1d6f01bf56747ad946f73f867af5770c4 Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Diffstat (limited to 'src/backends/backendsCommon/DynamicBackendUtils.cpp')
-rw-r--r--src/backends/backendsCommon/DynamicBackendUtils.cpp68
1 files changed, 67 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/DynamicBackendUtils.cpp b/src/backends/backendsCommon/DynamicBackendUtils.cpp
index 57a605608c..38ac5ad6af 100644
--- a/src/backends/backendsCommon/DynamicBackendUtils.cpp
+++ b/src/backends/backendsCommon/DynamicBackendUtils.cpp
@@ -281,7 +281,6 @@ std::vector<DynamicBackendPtr> DynamicBackendUtils::CreateDynamicBackends(const
{
BOOST_LOG_TRIVIAL(warning) << "Invalid dynamic backend object for the shared object file \""
<< sharedObject << "\"";
-
continue;
}
@@ -292,4 +291,71 @@ std::vector<DynamicBackendPtr> DynamicBackendUtils::CreateDynamicBackends(const
return dynamicBackends;
}
+void DynamicBackendUtils::RegisterDynamicBackends(const std::vector<DynamicBackendPtr>& dynamicBackends)
+{
+ // Get a reference of the backend registry
+ BackendRegistry& backendRegistry = BackendRegistryInstance();
+
+ // Register the dynamic backends in the backend registry
+ RegisterDynamicBackendsImpl(backendRegistry, dynamicBackends);
+}
+
+void DynamicBackendUtils::RegisterDynamicBackendsImpl(BackendRegistry& backendRegistry,
+ const std::vector<DynamicBackendPtr>& dynamicBackends)
+{
+ // Register the dynamic backends in the backend registry
+ for (const DynamicBackendPtr& dynamicBackend : dynamicBackends)
+ {
+ BackendId dynamicBackendId;
+ try
+ {
+ dynamicBackendId = dynamicBackend->GetBackendId();
+ }
+ catch (const RuntimeException& e)
+ {
+ BOOST_LOG_TRIVIAL(warning) << "Cannot register dynamic backend, "
+ << "an error has occurred when getting the backend id: " << e.what();
+ continue;
+ }
+ if (dynamicBackendId.IsEmpty() ||
+ dynamicBackendId.IsUndefined())
+ {
+ BOOST_LOG_TRIVIAL(warning) << "Cannot register dynamic backend, invalid backend id: " << dynamicBackendId;
+ continue;
+ }
+
+ // Check whether the dynamic backend is already registered
+ bool backendAlreadyRegistered = backendRegistry.IsBackendRegistered(dynamicBackendId);
+ if (backendAlreadyRegistered)
+ {
+ BOOST_LOG_TRIVIAL(warning) << "Cannot register dynamic backend \"" << dynamicBackendId
+ << "\": backend already registered";
+ continue;
+ }
+
+ // Get the dynamic backend factory function
+ BackendRegistry::FactoryFunction dynamicBackendFactoryFunction = nullptr;
+ try
+ {
+ dynamicBackendFactoryFunction = dynamicBackend->GetFactoryFunction();
+ }
+ catch (const RuntimeException& e)
+ {
+ BOOST_LOG_TRIVIAL(warning) << "Cannot register dynamic backend \"" << dynamicBackendId
+ << "\": an error has occurred when getting the backend factory function: "
+ << e.what();
+ continue;
+ }
+ if (dynamicBackendFactoryFunction == nullptr)
+ {
+ BOOST_LOG_TRIVIAL(warning) << "Cannot register dynamic backend \"" << dynamicBackendId
+ << "\": invalid backend factory function";
+ continue;
+ }
+
+ // Register the dynamic backend
+ backendRegistry.Register(dynamicBackendId, dynamicBackendFactoryFunction);
+ }
+}
+
} // namespace armnn