diff options
Diffstat (limited to 'src/backends/neon')
-rw-r--r-- | src/backends/neon/NeonWorkloadFactory.cpp | 13 | ||||
-rw-r--r-- | src/backends/neon/NeonWorkloadFactory.hpp | 2 |
2 files changed, 13 insertions, 2 deletions
diff --git a/src/backends/neon/NeonWorkloadFactory.cpp b/src/backends/neon/NeonWorkloadFactory.cpp index f0a9e76de1..c16d383554 100644 --- a/src/backends/neon/NeonWorkloadFactory.cpp +++ b/src/backends/neon/NeonWorkloadFactory.cpp @@ -3,6 +3,7 @@ // SPDX-License-Identifier: MIT // #include "NeonWorkloadFactory.hpp" +#include "NeonBackendId.hpp" #include <armnn/Utils.hpp> #include <backends/CpuTensorHandle.hpp> #include <Layer.hpp> @@ -25,11 +26,21 @@ namespace armnn { +namespace +{ +static const BackendId s_Id{NeonBackendId()}; +} + bool NeonWorkloadFactory::IsLayerSupported(const Layer& layer, Optional<DataType> dataType, std::string& outReasonIfUnsupported) { - return IWorkloadFactory::IsLayerSupported(Compute::CpuAcc, layer, dataType, outReasonIfUnsupported); + return IWorkloadFactory::IsLayerSupported(s_Id, layer, dataType, outReasonIfUnsupported); +} + +const BackendId& NeonWorkloadFactory::GetBackendId() const +{ + return s_Id; } #ifdef ARMCOMPUTENEON_ENABLED diff --git a/src/backends/neon/NeonWorkloadFactory.hpp b/src/backends/neon/NeonWorkloadFactory.hpp index d1dd2c85fe..030e982a20 100644 --- a/src/backends/neon/NeonWorkloadFactory.hpp +++ b/src/backends/neon/NeonWorkloadFactory.hpp @@ -20,7 +20,7 @@ class NeonWorkloadFactory : public IWorkloadFactory public: NeonWorkloadFactory(); - virtual Compute GetCompute() const override { return Compute::CpuAcc; } + const BackendId& GetBackendId() const override; static bool IsLayerSupported(const Layer& layer, Optional<DataType> dataType, |