src/backends/cl/ClBackend.cpp

   1 //
   2 // Copyright © 2017 Arm Ltd. All rights reserved.
   3 // SPDX-License-Identifier: MIT
   4 //
   5
   6 #include "ClBackend.hpp"
   7 #include "ClBackendId.hpp"
   8 #include "ClWorkloadFactory.hpp"
   9 #include "ClBackendContext.hpp"
  10 #include "ClLayerSupport.hpp"
  11 #include "ClTensorHandleFactory.hpp"
  12
  13 #include <aclCommon/BaseMemoryManager.hpp>
  14
  15 #include <backendsCommon/BackendRegistry.hpp>
  16 #include <backendsCommon/IBackendContext.hpp>
  17 #include <backendsCommon/IMemoryManager.hpp>
  18
  19 #include <Optimizer.hpp>
  20
  21 #include <arm_compute/runtime/CL/CLBufferAllocator.h>
  22
  23 #include <boost/polymorphic_pointer_cast.hpp>
  24
  25 namespace armnn
  26 {
  27
  28 const BackendId& ClBackend::GetIdStatic()
  29 {
  30     static const BackendId s_Id{ClBackendId()};
  31     return s_Id;
  32 }
  33
  34 IBackendInternal::IMemoryManagerUniquePtr ClBackend::CreateMemoryManager() const
  35 {
  36     return std::make_unique<ClMemoryManager>(std::make_unique<arm_compute::CLBufferAllocator>());
  37 }
  38
  39 IBackendInternal::IWorkloadFactoryPtr ClBackend::CreateWorkloadFactory(
  40     const IBackendInternal::IMemoryManagerSharedPtr& memoryManager) const
  41 {
  42     return std::make_unique<ClWorkloadFactory>(
  43         boost::polymorphic_pointer_downcast<ClMemoryManager>(memoryManager));
  44 }
  45
  46 IBackendInternal::IWorkloadFactoryPtr ClBackend::CreateWorkloadFactory(
  47     TensorHandleFactoryRegistry& registry) const
  48 {
  49     auto memoryManager = std::make_shared<ClMemoryManager>(std::make_unique<arm_compute::CLBufferAllocator>());
  50
  51     registry.RegisterMemoryManager(memoryManager);
  52
  53     return std::make_unique<ClWorkloadFactory>(
  54             boost::polymorphic_pointer_downcast<ClMemoryManager>(memoryManager));
  55 }
  56
  57 std::vector<ITensorHandleFactory::FactoryId> ClBackend::GetHandleFactoryPreferences() const
  58 {
  59     return std::vector<ITensorHandleFactory::FactoryId> {ClTensorHandleFactory::GetIdStatic()};
  60 }
  61
  62 void ClBackend::RegisterTensorHandleFactories(TensorHandleFactoryRegistry& registry)
  63 {
  64     auto mgr = std::make_shared<ClMemoryManager>(std::make_unique<arm_compute::CLBufferAllocator>());
  65
  66     registry.RegisterMemoryManager(mgr);
  67     registry.RegisterFactory(std::make_unique<ClTensorHandleFactory>(mgr));
  68 }
  69
  70 IBackendInternal::IBackendContextPtr
  71 ClBackend::CreateBackendContext(const IRuntime::CreationOptions& options) const
  72 {
  73     return IBackendContextPtr{new ClBackendContext{options}};
  74 }
  75
  76 IBackendInternal::Optimizations ClBackend::GetOptimizations() const
  77 {
  78     return Optimizations{};
  79 }
  80
  81 IBackendInternal::ILayerSupportSharedPtr ClBackend::GetLayerSupport() const
  82 {
  83     static ILayerSupportSharedPtr layerSupport{new ClLayerSupport};
  84     return layerSupport;
  85 }
  86
  87 OptimizationViews ClBackend::OptimizeSubgraphView(const SubgraphView& subgraph) const
  88 {
  89     OptimizationViews optimizationViews;
  90
  91     optimizationViews.AddUntouchedSubgraph(SubgraphView(subgraph));
  92
  93     return optimizationViews;
  94 }
  95
  96 } // namespace armnn