Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 1 | // |
| 2 | // Copyright © 2017 Arm Ltd. All rights reserved. |
| 3 | // SPDX-License-Identifier: MIT |
| 4 | // |
| 5 | |
| 6 | #include <reference/RefMemoryManager.hpp> |
| 7 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 8 | #include <doctest/doctest.h> |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 9 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 10 | TEST_SUITE("RefMemoryManagerTests") |
| 11 | { |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 12 | using namespace armnn; |
| 13 | using Pool = RefMemoryManager::Pool; |
| 14 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 15 | TEST_CASE("ManageOneThing") |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 16 | { |
| 17 | RefMemoryManager memoryManager; |
| 18 | |
| 19 | Pool* pool = memoryManager.Manage(10); |
| 20 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 21 | CHECK(pool); |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 22 | |
| 23 | memoryManager.Acquire(); |
| 24 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 25 | CHECK(memoryManager.GetPointer(pool) != nullptr); // Yields a valid pointer |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 26 | |
| 27 | memoryManager.Release(); |
| 28 | } |
| 29 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 30 | TEST_CASE("ManageTwoThings") |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 31 | { |
| 32 | RefMemoryManager memoryManager; |
| 33 | |
| 34 | Pool* pool1 = memoryManager.Manage(10); |
| 35 | Pool* pool2 = memoryManager.Manage(5); |
| 36 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 37 | CHECK(pool1); |
| 38 | CHECK(pool2); |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 39 | |
| 40 | memoryManager.Acquire(); |
| 41 | |
| 42 | void *p1 = memoryManager.GetPointer(pool1); |
| 43 | void *p2 = memoryManager.GetPointer(pool2); |
| 44 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 45 | CHECK(p1); |
| 46 | CHECK(p2); |
| 47 | CHECK(p1 != p2); |
Matthew Bentham | 7c1603a | 2019-06-21 17:22:23 +0100 | [diff] [blame] | 48 | |
| 49 | memoryManager.Release(); |
| 50 | } |
| 51 | |
Sadik Armagan | 1625efc | 2021-06-10 18:24:34 +0100 | [diff] [blame] | 52 | } |