10 #if defined(ARMCOMPUTENEON_ENABLED) || defined(ARMCOMPUTECL_ENABLED) 11 #include <arm_compute/runtime/MemoryGroup.h> 14 #if defined(ARMCOMPUTENEON_ENABLED) || defined(ARMCOMPUTECL_ENABLED) 15 #include <arm_compute/runtime/IAllocator.h> 16 #include <arm_compute/runtime/IMemoryGroup.h> 17 #include <arm_compute/runtime/MemoryManagerOnDemand.h> 20 #if defined(ARMCOMPUTECL_ENABLED) 21 #include <arm_compute/runtime/CL/CLTensorAllocator.h> 42 #if defined(ARMCOMPUTENEON_ENABLED) || defined(ARMCOMPUTECL_ENABLED) 57 virtual std::shared_ptr<arm_compute::IMemoryGroup>
58 CreateMemoryGroup(
const std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager) = 0;
62 #if defined(ARMCOMPUTENEON_ENABLED) 76 std::shared_ptr<arm_compute::IMemoryGroup>
77 CreateMemoryGroup(
const std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager)
override;
81 #if defined(ARMCOMPUTECL_ENABLED) 91 arm_compute::CLTensorAllocator::set_global_allocator(alloc.get());
96 std::shared_ptr<arm_compute::IMemoryGroup>
97 CreateMemoryGroup(
const std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager)
override;
std::shared_ptr< arm_compute::MemoryManagerOnDemand > CreateArmComputeMemoryManager(MemoryAffinity memoryAffinity)
ClMemoryManager(std::shared_ptr< arm_compute::IAllocator > alloc)
std::shared_ptr< arm_compute::MemoryManagerOnDemand > m_InterLayerMemoryMgr
std::shared_ptr< arm_compute::MemoryManagerOnDemand > m_IntraLayerMemoryMgr
Copyright (c) 2021 ARM Limited and Contributors.
std::shared_ptr< arm_compute::MemoryManagerOnDemand > & GetInterLayerManager()
NeonMemoryManager(std::unique_ptr< arm_compute::IAllocator > alloc, MemoryAffinity memoryAffinity)
virtual ~BaseMemoryManager()
virtual ~NeonMemoryManager()
std::shared_ptr< arm_compute::IMemoryGroup > m_InterLayerMemoryGroup
virtual ~ClMemoryManager()
std::shared_ptr< arm_compute::MemoryManagerOnDemand > & GetIntraLayerManager()
std::shared_ptr< arm_compute::IAllocator > m_Allocator
std::shared_ptr< arm_compute::IMemoryGroup > & GetInterLayerMemoryGroup()
virtual std::shared_ptr< arm_compute::IMemoryGroup > CreateMemoryGroup(const std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager)=0