From 17b12307edeaf488cfdf0cc3fa00b8f08293c93e Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Mon, 18 Jun 2018 18:13:51 +0100 Subject: COMPMID-1293: Handle aligned allocations Change-Id: I6e642c8cd968240f883c327464519e57e5d0c3e3 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/136088 Tested-by: Jenkins Reviewed-by: Anthony Barbier --- src/runtime/ITensorAllocator.cpp | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) (limited to 'src/runtime/ITensorAllocator.cpp') diff --git a/src/runtime/ITensorAllocator.cpp b/src/runtime/ITensorAllocator.cpp index 8294201384..087f324922 100644 --- a/src/runtime/ITensorAllocator.cpp +++ b/src/runtime/ITensorAllocator.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2016, 2017 ARM Limited. + * Copyright (c) 2016-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -31,13 +31,14 @@ using namespace arm_compute; ITensorAllocator::ITensorAllocator() - : _info() + : _info(), _alignment(0) { } -void ITensorAllocator::init(const TensorInfo &input) +void ITensorAllocator::init(const TensorInfo &input, size_t alignment) { - _info = input; + _info = input; + _alignment = alignment; } TensorInfo &ITensorAllocator::info() @@ -49,3 +50,8 @@ const TensorInfo &ITensorAllocator::info() const { return _info; } + +size_t ITensorAllocator::alignment() const +{ + return _alignment; +} -- cgit v1.2.1