diff options
Diffstat (limited to 'src/armnn/test/UnitTests.hpp')
| -rw-r--r-- | src/armnn/test/UnitTests.hpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnn/test/UnitTests.hpp b/src/armnn/test/UnitTests.hpp index e0505c9a9f..c15477bf19 100644 --- a/src/armnn/test/UnitTests.hpp +++ b/src/armnn/test/UnitTests.hpp @@ -64,7 +64,7 @@ void CompareTestResultIfSupported(const std::string& testName, const std::vector template<typename FactoryType, typename TFuncPtr, typename... Args> void RunTestFunction(const char* testName, TFuncPtr testFunction, Args... args) { - std::unique_ptr<armnn::Profiler> profiler = std::make_unique<armnn::Profiler>(); + std::unique_ptr<armnn::IProfiler> profiler = std::make_unique<armnn::IProfiler>(); armnn::ProfilerManager::GetInstance().RegisterProfiler(profiler.get()); auto memoryManager = WorkloadFactoryHelper<FactoryType>::GetMemoryManager(); @@ -80,7 +80,7 @@ void RunTestFunction(const char* testName, TFuncPtr testFunction, Args... args) template<typename FactoryType, typename TFuncPtr, typename... Args> void RunTestFunctionUsingTensorHandleFactory(const char* testName, TFuncPtr testFunction, Args... args) { - std::unique_ptr<armnn::Profiler> profiler = std::make_unique<armnn::Profiler>(); + std::unique_ptr<armnn::IProfiler> profiler = std::make_unique<armnn::IProfiler>(); armnn::ProfilerManager::GetInstance().RegisterProfiler(profiler.get()); auto memoryManager = WorkloadFactoryHelper<FactoryType>::GetMemoryManager(); |