aboutsummaryrefslogtreecommitdiff
path: root/src/armnn
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn')
-rw-r--r--src/armnn/Runtime.cpp6
-rw-r--r--src/armnn/Runtime.hpp8
2 files changed, 11 insertions, 3 deletions
diff --git a/src/armnn/Runtime.cpp b/src/armnn/Runtime.cpp
index 3505030273..ddfa6b4ffd 100644
--- a/src/armnn/Runtime.cpp
+++ b/src/armnn/Runtime.cpp
@@ -9,7 +9,6 @@
#include <backendsCommon/BackendRegistry.hpp>
#include <backendsCommon/IBackendContext.hpp>
#include <backendsCommon/DynamicBackendUtils.hpp>
-#include <backendsCommon/DynamicBackend.hpp>
#include <iostream>
@@ -253,7 +252,10 @@ void Runtime::LoadDynamicBackends(const std::string& overrideBackendPath)
std::vector<std::string> sharedObjects = DynamicBackendUtils::GetSharedObjects(backendPaths);
// Create a list of dynamic backends
- DynamicBackendUtils::CreateDynamicBackends(sharedObjects);
+ m_DynamicBackends = DynamicBackendUtils::CreateDynamicBackends(sharedObjects);
+
+ // Register the dynamic backends in the backend registry
+ DynamicBackendUtils::RegisterDynamicBackends(m_DynamicBackends);
}
} // namespace armnn
diff --git a/src/armnn/Runtime.hpp b/src/armnn/Runtime.hpp
index 624304ec0a..35684f1f78 100644
--- a/src/armnn/Runtime.hpp
+++ b/src/armnn/Runtime.hpp
@@ -6,11 +6,14 @@
#include "LoadedNetwork.hpp"
#include "DeviceSpec.hpp"
+
#include <armnn/INetwork.hpp>
#include <armnn/IRuntime.hpp>
#include <armnn/Tensor.hpp>
#include <armnn/BackendId.hpp>
+#include <backendsCommon/DynamicBackend.hpp>
+
#include <mutex>
#include <unordered_map>
@@ -100,6 +103,9 @@ private:
int m_NetworkIdCounter;
DeviceSpec m_DeviceSpec;
+
+ /// List of dynamic backends loaded in the runtime
+ std::vector<DynamicBackendPtr> m_DynamicBackends;
};
-}
+} // namespace armnn