diff options
Diffstat (limited to 'src/armnn')
-rw-r--r-- | src/armnn/Runtime.cpp | 6 | ||||
-rw-r--r-- | src/armnn/Runtime.hpp | 8 |
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 |