From 79141b9662547eeefb3ad533637223de40726e12 Mon Sep 17 00:00:00 2001
From: David Beck <david.beck@arm.com>
Date: Tue, 23 Oct 2018 16:09:36 +0100
Subject: IVGCVSW-2071 : remove GetCompute() from the WorkloadFactory interface

Change-Id: I44a9d26d1a5d876d381aee4c6450af62811d0dbb
---
 src/backends/reference/RefWorkloadFactory.cpp | 13 ++++++++++++-
 src/backends/reference/RefWorkloadFactory.hpp |  2 +-
 2 files changed, 13 insertions(+), 2 deletions(-)

(limited to 'src/backends/reference')

diff --git a/src/backends/reference/RefWorkloadFactory.cpp b/src/backends/reference/RefWorkloadFactory.cpp
index 783e5fba2e..864ffdbf4f 100644
--- a/src/backends/reference/RefWorkloadFactory.cpp
+++ b/src/backends/reference/RefWorkloadFactory.cpp
@@ -6,6 +6,7 @@
 #include <backends/MemCopyWorkload.hpp>
 #include <backends/MakeWorkloadHelper.hpp>
 #include "RefWorkloadFactory.hpp"
+#include "RefBackendId.hpp"
 #include "workloads/RefWorkloads.hpp"
 #include "Layer.hpp"
 
@@ -14,6 +15,11 @@
 namespace armnn
 {
 
+namespace
+{
+static const BackendId s_Id{RefBackendId()};
+}
+
 template <typename F32Workload, typename U8Workload, typename QueueDescriptorType>
 std::unique_ptr<IWorkload> RefWorkloadFactory::MakeWorkload(const QueueDescriptorType& descriptor,
     const WorkloadInfo& info) const
@@ -25,11 +31,16 @@ RefWorkloadFactory::RefWorkloadFactory()
 {
 }
 
+const BackendId& RefWorkloadFactory::GetBackendId() const
+{
+    return s_Id;
+}
+
 bool RefWorkloadFactory::IsLayerSupported(const Layer& layer,
                                           Optional<DataType> dataType,
                                           std::string& outReasonIfUnsupported)
 {
-    return IWorkloadFactory::IsLayerSupported(Compute::CpuRef, layer, dataType, outReasonIfUnsupported);
+    return IWorkloadFactory::IsLayerSupported(s_Id, layer, dataType, outReasonIfUnsupported);
 }
 
 std::unique_ptr<ITensorHandle> RefWorkloadFactory::CreateTensorHandle(const TensorInfo& tensorInfo) const
diff --git a/src/backends/reference/RefWorkloadFactory.hpp b/src/backends/reference/RefWorkloadFactory.hpp
index ef2e1abfaa..be0dafc159 100644
--- a/src/backends/reference/RefWorkloadFactory.hpp
+++ b/src/backends/reference/RefWorkloadFactory.hpp
@@ -33,7 +33,7 @@ public:
     explicit RefWorkloadFactory();
     virtual ~RefWorkloadFactory() {}
 
-    virtual Compute GetCompute() const override { return Compute::CpuRef; }
+    const BackendId& GetBackendId() const override;
 
     static bool IsLayerSupported(const Layer& layer,
                                  Optional<DataType> dataType,
-- 
cgit v1.2.1