diff options
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 |