diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2018-10-12 15:18:03 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:53 +0100 |
commit | f9aeef0e036df176699aa96d30d2ca8d7546534e (patch) | |
tree | 09ad918f7d2cffbca2c013c688332fa687b8d8ca /src/armnn/memory/BaseMemoryManager.hpp | |
parent | 3b278e9261bd0de67c82f7d6c36731f118124f52 (diff) | |
download | armnn-f9aeef0e036df176699aa96d30d2ca8d7546534e.tar.gz |
IVGCVSW-2006: Move ACL memory manager source code under aclCommon
Change-Id: Ie1c74a18de5c3dd1cd5285c222bd6327489c1508
Diffstat (limited to 'src/armnn/memory/BaseMemoryManager.hpp')
-rw-r--r-- | src/armnn/memory/BaseMemoryManager.hpp | 104 |
1 files changed, 0 insertions, 104 deletions
diff --git a/src/armnn/memory/BaseMemoryManager.hpp b/src/armnn/memory/BaseMemoryManager.hpp deleted file mode 100644 index c82eca6c1d..0000000000 --- a/src/armnn/memory/BaseMemoryManager.hpp +++ /dev/null @@ -1,104 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// -#pragma once - -#include <backends/WorkloadFactory.hpp> - -#ifdef ARMCOMPUTENEON_ENABLED -#include <arm_compute/runtime/MemoryGroup.h> -#endif - -#ifdef ARMCOMPUTECL_ENABLED -#include <arm_compute/runtime/CL/CLMemoryGroup.h> -#endif - -#if defined(ARMCOMPUTENEON_ENABLED) || defined(ARMCOMPUTECL_ENABLED) -#include <arm_compute/runtime/IAllocator.h> -#include <arm_compute/runtime/IMemoryGroup.h> -#include <arm_compute/runtime/MemoryManagerOnDemand.h> -#endif - -namespace armnn -{ - -class BaseMemoryManager -{ -public: - enum class MemoryAffinity - { - Buffer, - Offset - }; - - BaseMemoryManager() { } - virtual ~BaseMemoryManager() { } - -#if defined(ARMCOMPUTENEON_ENABLED) || defined(ARMCOMPUTECL_ENABLED) - - BaseMemoryManager(std::unique_ptr<arm_compute::IAllocator> alloc, MemoryAffinity memoryAffinity); - - std::shared_ptr<arm_compute::MemoryManagerOnDemand>& GetIntraLayerManager() { return m_IntraLayerMemoryMgr; } - std::shared_ptr<arm_compute::MemoryManagerOnDemand>& GetInterLayerManager() { return m_InterLayerMemoryMgr; } - std::shared_ptr<arm_compute::IMemoryGroup>& GetInterLayerMemoryGroup() { return m_InterLayerMemoryGroup; } - - void Finalize(); - void Acquire(); - void Release(); - -protected: - - std::unique_ptr<arm_compute::IAllocator> m_Allocator; - std::shared_ptr<arm_compute::MemoryManagerOnDemand> m_IntraLayerMemoryMgr; - std::shared_ptr<arm_compute::MemoryManagerOnDemand> m_InterLayerMemoryMgr; - std::shared_ptr<arm_compute::IMemoryGroup> m_InterLayerMemoryGroup; - - std::shared_ptr<arm_compute::MemoryManagerOnDemand> CreateArmComputeMemoryManager(MemoryAffinity memoryAffinity); - - virtual std::shared_ptr<arm_compute::IMemoryGroup> - CreateMemoryGroup(const std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager) = 0; - - void FinalizeMemoryManager(arm_compute::MemoryManagerOnDemand& memoryManager); -#endif -}; - -class NeonMemoryManager : public BaseMemoryManager -{ -public: - NeonMemoryManager() {} - virtual ~NeonMemoryManager() {} - -#ifdef ARMCOMPUTENEON_ENABLED - NeonMemoryManager(std::unique_ptr<arm_compute::IAllocator> alloc, MemoryAffinity memoryAffinity) - : BaseMemoryManager(std::move(alloc), memoryAffinity) - { - m_InterLayerMemoryGroup = CreateMemoryGroup(m_InterLayerMemoryMgr); - } - -protected: - virtual std::shared_ptr<arm_compute::IMemoryGroup> - CreateMemoryGroup(const std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager) override; -#endif -}; - -class ClMemoryManager : public BaseMemoryManager -{ -public: - ClMemoryManager() {} - virtual ~ClMemoryManager() {} - -#ifdef ARMCOMPUTECL_ENABLED - ClMemoryManager(std::unique_ptr<arm_compute::IAllocator> alloc) - : BaseMemoryManager(std::move(alloc), MemoryAffinity::Buffer) - { - m_InterLayerMemoryGroup = CreateMemoryGroup(m_InterLayerMemoryMgr); - } - -protected: - virtual std::shared_ptr<arm_compute::IMemoryGroup> - CreateMemoryGroup(const std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager) override; -#endif -}; - -} //namespace armnn
\ No newline at end of file |