aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/RefBackend.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/reference/RefBackend.hpp')
-rw-r--r--src/backends/reference/RefBackend.hpp18
1 files changed, 15 insertions, 3 deletions
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;
};