diff options
Diffstat (limited to 'src/backends/reference')
-rw-r--r-- | src/backends/reference/RefWorkloadFactory.cpp | 13 | ||||
-rw-r--r-- | src/backends/reference/RefWorkloadFactory.hpp | 2 |
2 files changed, 13 insertions, 2 deletions
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, |