aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/DeviceSpec.hpp
diff options
context:
space:
mode:
authorNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-12-09 17:24:41 +0000
committerNarumol Prangnawarat <narumol.prangnawarat@arm.com>2019-12-10 17:44:50 +0000
commit60a20fb62c992c4c6369bb7f532957f3e151f3d8 (patch)
treed9a33cbaeb33a24dddbf099d94f974493ed2cb60 /src/armnn/DeviceSpec.hpp
parented66d14ec0840282f74241dda46e02194278c3cc (diff)
downloadarmnn-60a20fb62c992c4c6369bb7f532957f3e151f3d8.tar.gz
IVGCVSW-4212 Example of standalone dynamic reference backend
* Add example to build dynamic reference backend * Add functions to clear dynamic backends * Fix the error when dynamic backend is not deregistered * Add DYARMNN_DYNAMIC_BACKEND_ENABLED to check disable empty dynamic backend when dynamic backend is enabled Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com> Change-Id: I1ef3a3f10ed6ca5ec18d0af04b007fc3bc71a3cb
Diffstat (limited to 'src/armnn/DeviceSpec.hpp')
-rw-r--r--src/armnn/DeviceSpec.hpp21
1 files changed, 20 insertions, 1 deletions
diff --git a/src/armnn/DeviceSpec.hpp b/src/armnn/DeviceSpec.hpp
index 32264706fd..703a4b123f 100644
--- a/src/armnn/DeviceSpec.hpp
+++ b/src/armnn/DeviceSpec.hpp
@@ -24,14 +24,33 @@ public:
return m_SupportedBackends;
}
- void AddSupportedBackends(const BackendIdSet& backendIds)
+ void AddSupportedBackends(const BackendIdSet& backendIds, bool isDynamic = false)
{
m_SupportedBackends.insert(backendIds.begin(), backendIds.end());
+ if (isDynamic)
+ {
+ m_DynamicBackends.insert(backendIds.begin(), backendIds.end());
+ }
+ }
+
+ void ClearDynamicBackends()
+ {
+ for (const auto& id : m_DynamicBackends)
+ {
+ m_SupportedBackends.erase(id);
+ }
+ m_DynamicBackends.clear();
+ }
+
+ const BackendIdSet& GetDynamicBackends() const
+ {
+ return m_DynamicBackends;
}
private:
DeviceSpec() = delete;
BackendIdSet m_SupportedBackends;
+ BackendIdSet m_DynamicBackends;
};
} // namespace armnn