diff options
Diffstat (limited to 'tests/MemoryStrategyBenchmark/MemoryStrategyBenchmark.cpp')
-rw-r--r-- | tests/MemoryStrategyBenchmark/MemoryStrategyBenchmark.cpp | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/tests/MemoryStrategyBenchmark/MemoryStrategyBenchmark.cpp b/tests/MemoryStrategyBenchmark/MemoryStrategyBenchmark.cpp index 5924757f21..1ffb7f4867 100644 --- a/tests/MemoryStrategyBenchmark/MemoryStrategyBenchmark.cpp +++ b/tests/MemoryStrategyBenchmark/MemoryStrategyBenchmark.cpp @@ -18,9 +18,11 @@ std::vector<TestBlock> testBlocks { + {"fsrcnn", fsrcnn}, {"inceptionv4", inceptionv4}, {"deeplabv3", deeplabv3}, {"deepspeechv1", deepspeechv1}, + {"mobilebert", mobilebert}, {"ssd_mobilenetv2", ssd_mobilenetv2}, {"resnetv2", resnetv2}, {"yolov3",yolov3} @@ -36,9 +38,16 @@ void PrintModels() std::cout << "\n"; } -size_t GetMinPossibleMemorySize(const std::vector<armnn::MemBlock> blocks) +size_t GetMinPossibleMemorySize(const std::vector<armnn::MemBlock>& blocks) { - std::vector<size_t> lifetimes(1000); + 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) |