diff options
Diffstat (limited to 'src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/TestMemBlocks.hpp')
-rw-r--r-- | src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/TestMemBlocks.hpp | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/TestMemBlocks.hpp b/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/TestMemBlocks.hpp new file mode 100644 index 0000000000..09369d1efa --- /dev/null +++ b/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/TestMemBlocks.hpp @@ -0,0 +1,43 @@ +// +// Copyright © 2021 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +size_t GetMinPossibleMemorySize(const std::vector<armnn::MemBlock>& blocks) +{ + unsigned int maxLifetime = 0; + for (auto& block: blocks) + { + maxLifetime = std::max(maxLifetime, block.m_EndOfLife); + } + maxLifetime++; + + std::vector<size_t> lifetimes(maxLifetime); + for (const auto& block : blocks) + { + for (auto lifetime = block.m_StartOfLife; lifetime <= block.m_EndOfLife; ++lifetime) + { + lifetimes[lifetime] += block.m_MemSize; + } + } + return *std::max_element(lifetimes.begin(), lifetimes.end()); +} + +// Generated from fsrcnn_720p.tflite +std::vector<armnn::MemBlock> fsrcnn +{ + { 0, 1, 691200, 0, 0 }, + { 1, 3, 7372800, 0, 1 }, + { 2, 5, 7372800, 0, 2 }, + { 3, 7, 1843200, 0, 3 }, + { 4, 9, 1843200, 0, 4 }, + { 5, 11, 1843200, 0, 5 }, + { 6, 13, 1843200, 0, 6 }, + { 7, 15, 1843200, 0, 7 }, + { 8, 17, 1843200, 0, 8 }, + { 9, 19, 7372800, 0, 9 }, + { 10, 21, 7372800, 0, 10 }, + { 11, 23, 2764800, 0, 11 }, + { 12, 25, 2764800, 0, 12 }, + { 13, 27, 2764800, 0, 13 } +};
\ No newline at end of file |