diff options
Diffstat (limited to 'src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test')
2 files changed, 29 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/CMakeLists.txt b/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/CMakeLists.txt index 3068b609f6..a82f718862 100644 --- a/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/CMakeLists.txt +++ b/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/CMakeLists.txt @@ -7,6 +7,7 @@ list(APPEND armnnMemoryOptimizationStrategiesUnitTests_sources ConstMemoryStrategyTests.cpp ValidatorStrategyTests.cpp SingleAxisPriorityListTests.cpp + MemoryOptimizerStrategyLibraryTests.cpp ) add_library(armnnMemoryOptimizationStrategiesUnitTests OBJECT ${armnnMemoryOptimizationStrategiesUnitTests_sources}) diff --git a/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/MemoryOptimizerStrategyLibraryTests.cpp b/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/MemoryOptimizerStrategyLibraryTests.cpp new file mode 100644 index 0000000000..482bc7d0bf --- /dev/null +++ b/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/test/MemoryOptimizerStrategyLibraryTests.cpp @@ -0,0 +1,28 @@ +// +// Copyright © 2021 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include <backendsCommon/memoryOptimizerStrategyLibrary/MemoryOptimizerStrategyLibrary.hpp> + +#include <doctest/doctest.h> + +using namespace armnn; + +TEST_SUITE("StrategyLibraryTestSuite") +{ + +TEST_CASE("StrategyLibraryTest") +{ + std::vector<std::string> strategyNames = GetMemoryOptimizerStrategyNames(); + CHECK(strategyNames.size() != 0); + for (const auto& strategyName: strategyNames) + { + auto strategy = GetMemoryOptimizerStrategy(strategyName); + CHECK(strategy); + CHECK(strategy->GetName() == strategyName); + } +} + +} + |