diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/runtime/ITensorAllocator.cpp | 14 | ||||
| -rw-r--r-- | src/runtime/TensorAllocator.cpp | 2 |
2 files changed, 11 insertions, 5 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; +} diff --git a/src/runtime/TensorAllocator.cpp b/src/runtime/TensorAllocator.cpp index 993a95b6c3..c84a2719d8 100644 --- a/src/runtime/TensorAllocator.cpp +++ b/src/runtime/TensorAllocator.cpp @@ -138,7 +138,7 @@ void TensorAllocator::allocate() if(_associated_memory_group == nullptr) { - _memory = Memory(std::make_shared<MemoryRegion>(info().total_size())); + _memory = Memory(std::make_shared<MemoryRegion>(info().total_size(), alignment())); } else { |