diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-12-14 17:11:20 +0000 |
|---|---|---|
| committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2018-12-14 19:45:00 +0000 |
| commit | 555f1c2241d6fa8c84926a72a0c54e4158817df4 (patch) | |
| tree | 0ec5da469c28559d8c5df9848cfa3ada6e0646e8 /arm_compute/runtime/MemoryGroupBase.h | |
| parent | b4af2c6738614850aaca3754904f0e8e3b17f0b2 (diff) | |
| download | ComputeLibrary-555f1c2241d6fa8c84926a72a0c54e4158817df4.tar.gz | |
COMPMID-1710: Account alignment for blob-base allocations
Change-Id: I290d33e25a5966d25a91df39ebc01c28bfa31f78
Reviewed-on: https://review.mlplatform.org/402
Reviewed-by: Anthony Barbier <Anthony.barbier@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/MemoryGroupBase.h')
| -rw-r--r-- | arm_compute/runtime/MemoryGroupBase.h | 7 |
1 files changed, 4 insertions, 3 deletions
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<TensorType>::manage(TensorType *obj) } template <typename TensorType> -inline void MemoryGroupBase<TensorType>::finalize_memory(TensorType *obj, IMemory &obj_memory, size_t size) +inline void MemoryGroupBase<TensorType>::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<TensorType>::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); } } |