aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/test/UnitTests.hpp
diff options
context:
space:
mode:
authorFrancis Murtagh <francis.murtagh@arm.com>2021-02-15 10:11:28 +0000
committerKeithARM <keith.davis@arm.com>2021-02-15 15:10:09 +0000
commit33199c25e5af1553e474a6f6eede07e888cd45ee (patch)
tree4d7980289479f83eb0cae81a460fb71a184f6dc9 /src/armnn/test/UnitTests.hpp
parent406463269f55a5baefb941b51e10f423f6d3250a (diff)
downloadarmnn-33199c25e5af1553e474a6f6eede07e888cd45ee.tar.gz
IVGCVSW-5675 Implement Pimpl Idiom for IProfiler (lower priority)
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com> Change-Id: If716f5f4e9b5433586b8a939d326830482da2f74
Diffstat (limited to 'src/armnn/test/UnitTests.hpp')
-rw-r--r--src/armnn/test/UnitTests.hpp4
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();