diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-06-18 18:13:51 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:53:20 +0000 |
| commit | 17b12307edeaf488cfdf0cc3fa00b8f08293c93e (patch) | |
| tree | d7025d84a03d9897b4811ade16a2a1952bdb09c4 /src/runtime/ITensorAllocator.cpp | |
| parent | 09b19129b65c5b8d1ca1c3851bab919bb9b7e1a1 (diff) | |
| download | ComputeLibrary-17b12307edeaf488cfdf0cc3fa00b8f08293c93e.tar.gz | |
COMPMID-1293: Handle aligned allocations
Change-Id: I6e642c8cd968240f883c327464519e57e5d0c3e3
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/136088
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/runtime/ITensorAllocator.cpp')
| -rw-r--r-- | src/runtime/ITensorAllocator.cpp | 14 |
1 files changed, 10 insertions, 4 deletions
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; +} |