aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/cl')
-rw-r--r--src/backends/cl/ClBackend.cpp22
-rw-r--r--src/backends/cl/ClBackend.hpp8
2 files changed, 23 insertions, 7 deletions
diff --git a/src/backends/cl/ClBackend.cpp b/src/backends/cl/ClBackend.cpp
index 5de9d31234..d185c15b72 100644
--- a/src/backends/cl/ClBackend.cpp
+++ b/src/backends/cl/ClBackend.cpp
@@ -4,11 +4,25 @@
//
#include "ClBackend.hpp"
+#include <backends/BackendRegistry.hpp>
+#include <boost/cast.hpp>
namespace armnn
{
-const std::string ClBackend::s_Id = "arm_compute_cl";
+namespace
+{
+static const std::string s_Id = "GpuAcc";
+
+static BackendRegistry::Helper g_RegisterHelper{
+ s_Id,
+ []()
+ {
+ return IBackendUniquePtr(new ClBackend, &ClBackend::Destroy);
+ }
+};
+
+}
const std::string& ClBackend::GetId() const
{
@@ -22,8 +36,12 @@ const ILayerSupport& ClBackend::GetLayerSupport() const
std::unique_ptr<IWorkloadFactory> ClBackend::CreateWorkloadFactory() const
{
- // TODO implement
return nullptr;
}
+void ClBackend::Destroy(IBackend* backend)
+{
+ delete boost::polymorphic_downcast<ClBackend*>(backend);
+}
+
} // namespace armnn \ No newline at end of file
diff --git a/src/backends/cl/ClBackend.hpp b/src/backends/cl/ClBackend.hpp
index 7e67f5e9ad..c43b6a6ce0 100644
--- a/src/backends/cl/ClBackend.hpp
+++ b/src/backends/cl/ClBackend.hpp
@@ -4,8 +4,7 @@
//
#pragma once
-#include "backends/IBackendInternal.hpp"
-
+#include <backends/IBackendInternal.hpp>
#include "ClLayerSupport.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:
ClLayerSupport m_LayerSupport;
};