From 555f1c2241d6fa8c84926a72a0c54e4158817df4 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 14 Dec 2018 17:11:20 +0000 Subject: COMPMID-1710: Account alignment for blob-base allocations Change-Id: I290d33e25a5966d25a91df39ebc01c28bfa31f78 Reviewed-on: https://review.mlplatform.org/402 Reviewed-by: Anthony Barbier Tested-by: Arm Jenkins --- arm_compute/runtime/MemoryGroupBase.h | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'arm_compute/runtime/MemoryGroupBase.h') diff --git a/arm_compute/runtime/MemoryGroupBase.h b/arm_compute/runtime/MemoryGroupBase.h index 0ceaa900c5..7dc18c8b4f 100644 --- a/arm_compute/runtime/MemoryGroupBase.h +++ b/arm_compute/runtime/MemoryGroupBase.h @@ -70,8 +70,9 @@ public: * @param[in, out] obj_memory Object's memory handling interface which can be used to alter the underlying memory * that is used by the object. * @param[in] size Size of memory to allocate + * @param[in] alignment (Optional) Alignment to use */ - void finalize_memory(TensorType *obj, IMemory &obj_memory, size_t size); + void finalize_memory(TensorType *obj, IMemory &obj_memory, size_t size, size_t alignment = 0); // Inherited methods overridden: void acquire() override; @@ -116,7 +117,7 @@ inline void MemoryGroupBase::manage(TensorType *obj) } template -inline void MemoryGroupBase::finalize_memory(TensorType *obj, IMemory &obj_memory, size_t size) +inline void MemoryGroupBase::finalize_memory(TensorType *obj, IMemory &obj_memory, size_t size, size_t alignment) { // TODO (geopin01) : Check size (track size in MemoryMappings) // Check if existing mapping is valid @@ -125,7 +126,7 @@ inline void MemoryGroupBase::finalize_memory(TensorType *obj, IMemor if(_memory_manager && _mappings.empty()) { ARM_COMPUTE_ERROR_ON(!_memory_manager->lifetime_manager()); - _memory_manager->lifetime_manager()->end_lifetime(obj, obj_memory, size); + _memory_manager->lifetime_manager()->end_lifetime(obj, obj_memory, size, alignment); } } -- cgit v1.2.1