aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/neon')
-rw-r--r--src/backends/neon/NeonBackend.cpp23
-rw-r--r--src/backends/neon/NeonBackend.hpp8
2 files changed, 24 insertions, 7 deletions
diff --git a/src/backends/neon/NeonBackend.cpp b/src/backends/neon/NeonBackend.cpp
index 1137145aa2..b4f1897704 100644
--- a/src/backends/neon/NeonBackend.cpp
+++ b/src/backends/neon/NeonBackend.cpp
@@ -4,11 +4,26 @@
//
#include "NeonBackend.hpp"
+#include <backends/BackendRegistry.hpp>
+#include <boost/cast.hpp>
namespace armnn
{
-const std::string NeonBackend::s_Id = "arm_compute_neon";
+namespace
+{
+
+static const std::string s_Id = "CpuAcc";
+
+static BackendRegistry::Helper g_RegisterHelper{
+ s_Id,
+ []()
+ {
+ return IBackendUniquePtr(new NeonBackend, &NeonBackend::Destroy);
+ }
+};
+
+}
const std::string& NeonBackend::GetId() const
{
@@ -22,8 +37,12 @@ const ILayerSupport& NeonBackend::GetLayerSupport() const
std::unique_ptr<IWorkloadFactory> NeonBackend::CreateWorkloadFactory() const
{
- // TODO implement
return nullptr;
}
+void NeonBackend::Destroy(IBackend* backend)
+{
+ delete boost::polymorphic_downcast<NeonBackend*>(backend);
+}
+
} // namespace armnn \ No newline at end of file
diff --git a/src/backends/neon/NeonBackend.hpp b/src/backends/neon/NeonBackend.hpp
index 86fccf0e51..5d4bd5dfcc 100644
--- a/src/backends/neon/NeonBackend.hpp
+++ b/src/backends/neon/NeonBackend.hpp
@@ -4,8 +4,7 @@
//
#pragma once
-#include "backends/IBackendInternal.hpp"
-
+#include <backends/IBackendInternal.hpp>
#include "NeonLayerSupport.hpp"
namespace armnn
@@ -23,10 +22,9 @@ public:
std::unique_ptr<IWorkloadFactory> CreateWorkloadFactory() const override;
-private:
- static const std::string s_Id;
+ static void Destroy(IBackend* backend);
- // TODO initialize
+private:
NeonLayerSupport m_LayerSupport;
};