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