IVGCVSW-5855 Refactor the reporting of capabilities from backends

Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I05fc331a8e91bdcb6b8a2f32cfb555060fc5d797

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;
 };