diff options
| author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-09-13 13:37:03 +0100 |
|---|---|---|
| committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-09-16 10:09:26 +0000 |
| commit | d9f7c8ba3949823a623b407f4bd80d120ca0b5be (patch) | |
| tree | 9835b10fced78f38ad89f0d9c08050d4ed0c3fa6 /src/backends/reference/RefTensorHandle.cpp | |
| parent | 46d1c62aa6c8ef634311257c6adeb6775cc8e0f8 (diff) | |
| download | armnn-d9f7c8ba3949823a623b407f4bd80d120ca0b5be.tar.gz | |
IVGCVSW-3858 Fix RefTensorHandleTests on Raspberry Pi
* Fix alignment check to use sizeof(size_t) instead of a hard-coded value
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I092c4464c6cecb2403da9b7744b68ad063ddbad1
Diffstat (limited to 'src/backends/reference/RefTensorHandle.cpp')
| -rw-r--r-- | src/backends/reference/RefTensorHandle.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/reference/RefTensorHandle.cpp b/src/backends/reference/RefTensorHandle.cpp index 42ac7f0826..84a74edc1d 100644 --- a/src/backends/reference/RefTensorHandle.cpp +++ b/src/backends/reference/RefTensorHandle.cpp @@ -110,8 +110,9 @@ bool RefTensorHandle::Import(void* memory, MemorySource source) { if (source == MemorySource::Malloc) { - // Checks the 16 byte memory alignment. - if (reinterpret_cast<uint64_t>(memory) % 16) + // Check memory alignment + constexpr uintptr_t alignment = sizeof(size_t); + if (reinterpret_cast<uintptr_t>(memory) % alignment) { if (m_Imported) { |