diff options
author | Finn Williams <Finn.Williams@arm.com> | 2021-05-26 18:38:12 +0100 |
---|---|---|
committer | Finn Williams <Finn.Williams@arm.com> | 2021-06-09 17:18:10 +0100 |
commit | b9af86ea42568ade799ee5529137e4756977b6c6 (patch) | |
tree | 261003078fd2191b22ee7465e07668cbed666553 /src/backends/reference | |
parent | 5b1bcc93820b442bc4035c1e030a8d4a0983df91 (diff) | |
download | armnn-b9af86ea42568ade799ee5529137e4756977b6c6.tar.gz |
IVGCVSW-5855 Refactor the reporting of capabilities from backends
Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I05fc331a8e91bdcb6b8a2f32cfb555060fc5d797
Diffstat (limited to 'src/backends/reference')
-rw-r--r-- | src/backends/reference/RefBackend.cpp | 4 | ||||
-rw-r--r-- | src/backends/reference/RefBackend.hpp | 18 |
2 files changed, 17 insertions, 5 deletions
diff --git a/src/backends/reference/RefBackend.cpp b/src/backends/reference/RefBackend.cpp index 85c6a2e1e4..c9f164e0c9 100644 --- a/src/backends/reference/RefBackend.cpp +++ b/src/backends/reference/RefBackend.cpp @@ -71,8 +71,8 @@ IBackendInternal::ILayerSupportSharedPtr RefBackend::GetLayerSupport() const bool RefBackend::HasCapability(BackendCapability capabilityClass) const { - auto search = cpuRefCapabilities.find(capabilityClass); - if (search != cpuRefCapabilities.end()) + auto search = oldCpuRefCapabilities.find(capabilityClass); + if (search != oldCpuRefCapabilities.end()) { return true; } diff --git a/src/backends/reference/RefBackend.hpp b/src/backends/reference/RefBackend.hpp index 3143173061..441f4ebdf4 100644 --- a/src/backends/reference/RefBackend.hpp +++ b/src/backends/reference/RefBackend.hpp @@ -8,11 +8,18 @@ namespace armnn { - -const std::set<armnn::BackendCapability> cpuRefCapabilities { - armnn::BackendCapability::NonConstWeights, +// add new capabilities here.. +const BackendCapabilities cpuRefCapabilities("CpuRef", + { + {"NonConstWeights", true}, + {"AsyncExecution", true} + }); + +const std::set<armnn::BackendCapability> oldCpuRefCapabilities { + armnn::BackendCapability::NonConstWeights, }; + class RefBackend : public IBackendInternal { public: @@ -44,6 +51,11 @@ public: void RegisterTensorHandleFactories(class TensorHandleFactoryRegistry& registry) override; + BackendCapabilities GetCapabilities() const override + { + return cpuRefCapabilities; + }; + bool HasCapability(BackendCapability capabilityClass) const override; }; |