aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon')
-rw-r--r--src/backends/neon/NeonWorkloadFactory.cpp13
-rw-r--r--src/backends/neon/NeonWorkloadFactory.hpp2
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,