Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 1 | // |
Colm Donelan | a98e79a | 2022-12-06 21:32:29 +0000 | [diff] [blame] | 2 | // Copyright © 2017,2022 Arm Ltd and Contributors. All rights reserved. |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 3 | // SPDX-License-Identifier: MIT |
| 4 | // |
Colm Donelan | a98e79a | 2022-12-06 21:32:29 +0000 | [diff] [blame] | 5 | |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 6 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 7 | #include <doctest/doctest.h> |
Colm Donelan | a98e79a | 2022-12-06 21:32:29 +0000 | [diff] [blame] | 8 | #include <armnn/BackendId.hpp> |
| 9 | #include <armnn/INetwork.hpp> |
| 10 | #include <armnn/Tensor.hpp> |
| 11 | #include <armnn/Types.hpp> |
| 12 | #include <armnn/backends/ITensorHandle.hpp> |
| 13 | #include <armnn/backends/ITensorHandleFactory.hpp> |
| 14 | #include <armnn/backends/TensorHandle.hpp> |
| 15 | #include <armnn/utility/Assert.hpp> |
| 16 | #include <reference/RefTensorHandle.hpp> |
| 17 | #include <reference/RefTensorHandleFactory.hpp> |
| 18 | #include <reference/RefMemoryManager.hpp> |
| 19 | #include <memory> |
| 20 | #include <vector> |
| 21 | |
| 22 | namespace armnn |
| 23 | { |
| 24 | class Exception; |
| 25 | class NullPointerException; |
| 26 | } |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 27 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 28 | TEST_SUITE("RefTensorHandleTests") |
| 29 | { |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 30 | using namespace armnn; |
| 31 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 32 | TEST_CASE("AcquireAndRelease") |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 33 | { |
| 34 | std::shared_ptr<RefMemoryManager> memoryManager = std::make_shared<RefMemoryManager>(); |
| 35 | |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 36 | TensorInfo info({ 1, 1, 1, 1 }, DataType::Float32); |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 37 | RefTensorHandle handle(info, memoryManager); |
| 38 | |
| 39 | handle.Manage(); |
| 40 | handle.Allocate(); |
| 41 | |
| 42 | memoryManager->Acquire(); |
| 43 | { |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 44 | float* buffer = reinterpret_cast<float*>(handle.Map()); |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 45 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 46 | CHECK(buffer != nullptr); // Yields a valid pointer |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 47 | |
| 48 | buffer[0] = 2.5f; |
| 49 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 50 | CHECK(buffer[0] == 2.5f); // Memory is writable and readable |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 51 | |
| 52 | } |
| 53 | memoryManager->Release(); |
| 54 | |
| 55 | memoryManager->Acquire(); |
| 56 | { |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 57 | float* buffer = reinterpret_cast<float*>(handle.Map()); |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 58 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 59 | CHECK(buffer != nullptr); // Yields a valid pointer |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 60 | |
| 61 | buffer[0] = 3.5f; |
| 62 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 63 | CHECK(buffer[0] == 3.5f); // Memory is writable and readable |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 64 | } |
| 65 | memoryManager->Release(); |
| 66 | } |
| 67 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 68 | TEST_CASE("RefTensorHandleFactoryMemoryManaged") |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 69 | { |
| 70 | std::shared_ptr<RefMemoryManager> memoryManager = std::make_shared<RefMemoryManager>(); |
| 71 | RefTensorHandleFactory handleFactory(memoryManager); |
| 72 | TensorInfo info({ 1, 1, 2, 1 }, DataType::Float32); |
| 73 | |
| 74 | // create TensorHandle with memory managed |
| 75 | auto handle = handleFactory.CreateTensorHandle(info, true); |
| 76 | handle->Manage(); |
| 77 | handle->Allocate(); |
| 78 | |
| 79 | memoryManager->Acquire(); |
| 80 | { |
| 81 | float* buffer = reinterpret_cast<float*>(handle->Map()); |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 82 | CHECK(buffer != nullptr); // Yields a valid pointer |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 83 | buffer[0] = 1.5f; |
| 84 | buffer[1] = 2.5f; |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 85 | CHECK(buffer[0] == 1.5f); // Memory is writable and readable |
| 86 | CHECK(buffer[1] == 2.5f); // Memory is writable and readable |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 87 | } |
| 88 | memoryManager->Release(); |
| 89 | |
| 90 | memoryManager->Acquire(); |
| 91 | { |
| 92 | float* buffer = reinterpret_cast<float*>(handle->Map()); |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 93 | CHECK(buffer != nullptr); // Yields a valid pointer |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 94 | buffer[0] = 3.5f; |
| 95 | buffer[1] = 4.5f; |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 96 | CHECK(buffer[0] == 3.5f); // Memory is writable and readable |
| 97 | CHECK(buffer[1] == 4.5f); // Memory is writable and readable |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 98 | } |
| 99 | memoryManager->Release(); |
| 100 | |
| 101 | float testPtr[2] = { 2.5f, 5.5f }; |
Matthew Bentham | 6b5f674 | 2022-11-23 18:17:48 +0000 | [diff] [blame^] | 102 | // Check import overlays contents |
| 103 | CHECK(handle->Import(static_cast<void*>(testPtr), MemorySource::Malloc)); |
| 104 | { |
| 105 | float* buffer = reinterpret_cast<float*>(handle->Map()); |
| 106 | CHECK(buffer != nullptr); // Yields a valid pointer |
| 107 | CHECK(buffer[0] == 2.5f); // Memory is writable and readable |
| 108 | CHECK(buffer[1] == 5.5f); // Memory is writable and readable |
| 109 | } |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 110 | } |
| 111 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 112 | TEST_CASE("RefTensorHandleFactoryImport") |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 113 | { |
| 114 | std::shared_ptr<RefMemoryManager> memoryManager = std::make_shared<RefMemoryManager>(); |
| 115 | RefTensorHandleFactory handleFactory(memoryManager); |
| 116 | TensorInfo info({ 1, 1, 2, 1 }, DataType::Float32); |
| 117 | |
| 118 | // create TensorHandle without memory managed |
| 119 | auto handle = handleFactory.CreateTensorHandle(info, false); |
| 120 | handle->Manage(); |
| 121 | handle->Allocate(); |
| 122 | memoryManager->Acquire(); |
| 123 | |
Matthew Bentham | 6b5f674 | 2022-11-23 18:17:48 +0000 | [diff] [blame^] | 124 | // Check storage has been allocated |
| 125 | void* unmanagedStorage = handle->Map(); |
| 126 | CHECK(unmanagedStorage != nullptr); |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 127 | |
Matthew Bentham | 6b5f674 | 2022-11-23 18:17:48 +0000 | [diff] [blame^] | 128 | // Check importing overlays the storage |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 129 | float testPtr[2] = { 2.5f, 5.5f }; |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 130 | CHECK(handle->Import(static_cast<void*>(testPtr), MemorySource::Malloc)); |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 131 | float* buffer = reinterpret_cast<float*>(handle->Map()); |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 132 | CHECK(buffer != nullptr); // Yields a valid pointer after import |
| 133 | CHECK(buffer == testPtr); // buffer is pointing to testPtr |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 134 | // Memory is writable and readable with correct value |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 135 | CHECK(buffer[0] == 2.5f); |
| 136 | CHECK(buffer[1] == 5.5f); |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 137 | buffer[0] = 3.5f; |
| 138 | buffer[1] = 10.0f; |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 139 | CHECK(buffer[0] == 3.5f); |
| 140 | CHECK(buffer[1] == 10.0f); |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 141 | memoryManager->Release(); |
| 142 | } |
| 143 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 144 | TEST_CASE("RefTensorHandleImport") |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 145 | { |
| 146 | TensorInfo info({ 1, 1, 2, 1 }, DataType::Float32); |
Matthew Bentham | c30abd8 | 2022-11-23 12:11:32 +0000 | [diff] [blame] | 147 | RefTensorHandle handle(info); |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 148 | |
| 149 | handle.Manage(); |
| 150 | handle.Allocate(); |
| 151 | |
Matthew Bentham | 6b5f674 | 2022-11-23 18:17:48 +0000 | [diff] [blame^] | 152 | // Check unmanaged memory allocated |
| 153 | CHECK(handle.Map()); |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 154 | |
| 155 | float testPtr[2] = { 2.5f, 5.5f }; |
Matthew Bentham | 6b5f674 | 2022-11-23 18:17:48 +0000 | [diff] [blame^] | 156 | // Check imoport overlays the unamaged memory |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 157 | CHECK(handle.Import(static_cast<void*>(testPtr), MemorySource::Malloc)); |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 158 | float* buffer = reinterpret_cast<float*>(handle.Map()); |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 159 | CHECK(buffer != nullptr); // Yields a valid pointer after import |
| 160 | CHECK(buffer == testPtr); // buffer is pointing to testPtr |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 161 | // Memory is writable and readable with correct value |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 162 | CHECK(buffer[0] == 2.5f); |
| 163 | CHECK(buffer[1] == 5.5f); |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 164 | buffer[0] = 3.5f; |
| 165 | buffer[1] = 10.0f; |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 166 | CHECK(buffer[0] == 3.5f); |
| 167 | CHECK(buffer[1] == 10.0f); |
Narumol Prangnawarat | 3b90af6 | 2020-06-26 11:00:21 +0100 | [diff] [blame] | 168 | } |
| 169 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 170 | TEST_CASE("RefTensorHandleGetCapabilities") |
Narumol Prangnawarat | d656877 | 2020-07-22 12:46:51 +0100 | [diff] [blame] | 171 | { |
| 172 | std::shared_ptr<RefMemoryManager> memoryManager = std::make_shared<RefMemoryManager>(); |
| 173 | RefTensorHandleFactory handleFactory(memoryManager); |
| 174 | |
| 175 | // Builds up the structure of the network. |
| 176 | INetworkPtr net(INetwork::Create()); |
| 177 | IConnectableLayer* input = net->AddInputLayer(0); |
| 178 | IConnectableLayer* output = net->AddOutputLayer(0); |
| 179 | input->GetOutputSlot(0).Connect(output->GetInputSlot(0)); |
| 180 | |
| 181 | std::vector<Capability> capabilities = handleFactory.GetCapabilities(input, |
| 182 | output, |
| 183 | CapabilityClass::PaddingRequired); |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 184 | CHECK(capabilities.empty()); |
Narumol Prangnawarat | d656877 | 2020-07-22 12:46:51 +0100 | [diff] [blame] | 185 | } |
| 186 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 187 | TEST_CASE("RefTensorHandleSupportsInPlaceComputation") |
Sadik Armagan | ab3bd4d | 2020-08-25 11:48:00 +0100 | [diff] [blame] | 188 | { |
| 189 | std::shared_ptr<RefMemoryManager> memoryManager = std::make_shared<RefMemoryManager>(); |
| 190 | RefTensorHandleFactory handleFactory(memoryManager); |
| 191 | |
| 192 | // RefTensorHandleFactory does not support InPlaceComputation |
| 193 | ARMNN_ASSERT(!(handleFactory.SupportsInPlaceComputation())); |
| 194 | } |
| 195 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 196 | TEST_CASE("TestManagedConstTensorHandle") |
Francis Murtagh | 4af5616 | 2021-04-20 16:37:55 +0100 | [diff] [blame] | 197 | { |
| 198 | // Initialize arguments |
| 199 | void* mem = nullptr; |
| 200 | TensorInfo info; |
| 201 | |
James Conroy | 1f58f03 | 2021-04-27 17:13:27 +0100 | [diff] [blame] | 202 | // Use PassthroughTensor as others are abstract |
| 203 | auto passThroughHandle = std::make_shared<PassthroughTensorHandle>(info, mem); |
Francis Murtagh | 4af5616 | 2021-04-20 16:37:55 +0100 | [diff] [blame] | 204 | |
| 205 | // Test managed handle is initialized with m_Mapped unset and once Map() called its set |
| 206 | ManagedConstTensorHandle managedHandle(passThroughHandle); |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 207 | CHECK(!managedHandle.IsMapped()); |
Francis Murtagh | 4af5616 | 2021-04-20 16:37:55 +0100 | [diff] [blame] | 208 | managedHandle.Map(); |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 209 | CHECK(managedHandle.IsMapped()); |
Francis Murtagh | 4af5616 | 2021-04-20 16:37:55 +0100 | [diff] [blame] | 210 | |
| 211 | // Test it can then be unmapped |
| 212 | managedHandle.Unmap(); |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 213 | CHECK(!managedHandle.IsMapped()); |
Francis Murtagh | 4af5616 | 2021-04-20 16:37:55 +0100 | [diff] [blame] | 214 | |
| 215 | // Test member function |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 216 | CHECK(managedHandle.GetTensorInfo() == info); |
Francis Murtagh | 4af5616 | 2021-04-20 16:37:55 +0100 | [diff] [blame] | 217 | |
| 218 | // Test that nullptr tensor handle doesn't get mapped |
| 219 | ManagedConstTensorHandle managedHandleNull(nullptr); |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 220 | CHECK(!managedHandleNull.IsMapped()); |
| 221 | CHECK_THROWS_AS(managedHandleNull.Map(), armnn::Exception); |
| 222 | CHECK(!managedHandleNull.IsMapped()); |
Francis Murtagh | 4af5616 | 2021-04-20 16:37:55 +0100 | [diff] [blame] | 223 | |
| 224 | // Check Unmap() when m_Mapped already false |
| 225 | managedHandleNull.Unmap(); |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 226 | CHECK(!managedHandleNull.IsMapped()); |
Francis Murtagh | 4af5616 | 2021-04-20 16:37:55 +0100 | [diff] [blame] | 227 | } |
| 228 | |
Ferran Balaguer | c33882d | 2019-08-21 13:59:13 +0100 | [diff] [blame] | 229 | #if !defined(__ANDROID__) |
| 230 | // Only run these tests on non Android platforms |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 231 | TEST_CASE("CheckSourceType") |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 232 | { |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 233 | TensorInfo info({1}, DataType::Float32); |
Matthew Bentham | c30abd8 | 2022-11-23 12:11:32 +0000 | [diff] [blame] | 234 | RefTensorHandle handle(info); |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 235 | |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 236 | int* testPtr = new int(4); |
| 237 | |
| 238 | // Not supported |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 239 | CHECK(!handle.Import(static_cast<void *>(testPtr), MemorySource::DmaBuf)); |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 240 | |
| 241 | // Not supported |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 242 | CHECK(!handle.Import(static_cast<void *>(testPtr), MemorySource::DmaBufProtected)); |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 243 | |
| 244 | // Supported |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 245 | CHECK(handle.Import(static_cast<void *>(testPtr), MemorySource::Malloc)); |
Ferran Balaguer | 1cd451c | 2019-08-22 14:09:44 +0100 | [diff] [blame] | 246 | |
| 247 | delete testPtr; |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 248 | } |
| 249 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 250 | TEST_CASE("ReusePointer") |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 251 | { |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 252 | TensorInfo info({1}, DataType::Float32); |
Matthew Bentham | c30abd8 | 2022-11-23 12:11:32 +0000 | [diff] [blame] | 253 | RefTensorHandle handle(info); |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 254 | |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 255 | int* testPtr = new int(4); |
| 256 | |
| 257 | handle.Import(static_cast<void *>(testPtr), MemorySource::Malloc); |
| 258 | |
| 259 | // Reusing previously Imported pointer |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 260 | CHECK(handle.Import(static_cast<void *>(testPtr), MemorySource::Malloc)); |
Ferran Balaguer | 1cd451c | 2019-08-22 14:09:44 +0100 | [diff] [blame] | 261 | |
| 262 | delete testPtr; |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 263 | } |
| 264 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 265 | TEST_CASE("MisalignedPointer") |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 266 | { |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 267 | TensorInfo info({2}, DataType::Float32); |
Matthew Bentham | c30abd8 | 2022-11-23 12:11:32 +0000 | [diff] [blame] | 268 | RefTensorHandle handle(info); |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 269 | |
Aron Virginas-Tar | d9f7c8b | 2019-09-13 13:37:03 +0100 | [diff] [blame] | 270 | // Allocate a 2 int array |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 271 | int* testPtr = new int[2]; |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 272 | |
Aron Virginas-Tar | d9f7c8b | 2019-09-13 13:37:03 +0100 | [diff] [blame] | 273 | // Increment pointer by 1 byte |
| 274 | void* misalignedPtr = static_cast<void*>(reinterpret_cast<char*>(testPtr) + 1); |
| 275 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 276 | CHECK(!handle.Import(misalignedPtr, MemorySource::Malloc)); |
Ferran Balaguer | bfeb271 | 2019-08-07 15:14:56 +0100 | [diff] [blame] | 277 | |
| 278 | delete[] testPtr; |
| 279 | } |
| 280 | |
Nikhil Raj | 53e0659 | 2022-01-05 16:04:08 +0000 | [diff] [blame] | 281 | TEST_CASE("CheckCanBeImported") |
| 282 | { |
| 283 | TensorInfo info({1}, DataType::Float32); |
Matthew Bentham | c30abd8 | 2022-11-23 12:11:32 +0000 | [diff] [blame] | 284 | RefTensorHandle handle(info); |
Nikhil Raj | 53e0659 | 2022-01-05 16:04:08 +0000 | [diff] [blame] | 285 | |
| 286 | int* testPtr = new int(4); |
| 287 | |
| 288 | // Not supported |
| 289 | CHECK(!handle.CanBeImported(static_cast<void *>(testPtr), MemorySource::DmaBuf)); |
| 290 | |
| 291 | // Supported |
| 292 | CHECK(handle.CanBeImported(static_cast<void *>(testPtr), MemorySource::Malloc)); |
| 293 | |
| 294 | delete testPtr; |
| 295 | |
| 296 | } |
| 297 | |
| 298 | TEST_CASE("MisalignedCanBeImported") |
| 299 | { |
| 300 | TensorInfo info({2}, DataType::Float32); |
Matthew Bentham | c30abd8 | 2022-11-23 12:11:32 +0000 | [diff] [blame] | 301 | RefTensorHandle handle(info); |
Nikhil Raj | 53e0659 | 2022-01-05 16:04:08 +0000 | [diff] [blame] | 302 | |
| 303 | // Allocate a 2 int array |
| 304 | int* testPtr = new int[2]; |
| 305 | |
| 306 | // Increment pointer by 1 byte |
| 307 | void* misalignedPtr = static_cast<void*>(reinterpret_cast<char*>(testPtr) + 1); |
| 308 | |
| 309 | CHECK(!handle.Import(misalignedPtr, MemorySource::Malloc)); |
| 310 | |
| 311 | delete[] testPtr; |
| 312 | } |
| 313 | |
Ferran Balaguer | c33882d | 2019-08-21 13:59:13 +0100 | [diff] [blame] | 314 | #endif |
| 315 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 316 | } |