diff options
author | Finn Williams <finn.williams@arm.com> | 2021-11-08 15:22:45 +0000 |
---|---|---|
committer | David Monahan <david.monahan@arm.com> | 2021-11-08 18:11:37 +0000 |
commit | b03e8ffd1a895b680dca1ce90c049fa7a9a40cb0 (patch) | |
tree | 03a97f1cb5d4d1d88324c059f782d840b7b0522d /src/backends/backendsCommon/memoryOptimizerStrategyLibrary/MemoryOptimizerStrategyFactory.hpp | |
parent | f8fb46df602c72d62defe82e3283b33f9eeccdd3 (diff) | |
download | armnn-b03e8ffd1a895b680dca1ce90c049fa7a9a40cb0.tar.gz |
Fix MemoryOptimizerStrategyLibrary search
Signed-off-by: Finn Williams <finn.williams@arm.com>
Change-Id: I4ca8d9196abd0e116d420a36c780e39edbca0eb3
Diffstat (limited to 'src/backends/backendsCommon/memoryOptimizerStrategyLibrary/MemoryOptimizerStrategyFactory.hpp')
-rw-r--r-- | src/backends/backendsCommon/memoryOptimizerStrategyLibrary/MemoryOptimizerStrategyFactory.hpp | 14 |
1 files changed, 8 insertions, 6 deletions
diff --git a/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/MemoryOptimizerStrategyFactory.hpp b/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/MemoryOptimizerStrategyFactory.hpp index aff0995266..7b04f442d2 100644 --- a/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/MemoryOptimizerStrategyFactory.hpp +++ b/src/backends/backendsCommon/memoryOptimizerStrategyLibrary/MemoryOptimizerStrategyFactory.hpp @@ -12,17 +12,19 @@ namespace armnn { -class MemoryOptimizerStrategyFactory +struct IMemoryOptimizerStrategyFactory { -public: - MemoryOptimizerStrategyFactory() {} + virtual ~IMemoryOptimizerStrategyFactory() = default; + virtual std::unique_ptr<IMemoryOptimizerStrategy> CreateMemoryOptimizerStrategy() = 0; +}; - template <typename T> - std::unique_ptr<IMemoryOptimizerStrategy> CreateMemoryOptimizerStrategy() +template <typename T> +struct StrategyFactory : public IMemoryOptimizerStrategyFactory +{ + std::unique_ptr<IMemoryOptimizerStrategy> CreateMemoryOptimizerStrategy() override { return std::make_unique<T>(); } - }; } // namespace armnn
\ No newline at end of file |