diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2021-02-15 10:11:28 +0000 |
---|---|---|
committer | KeithARM <keith.davis@arm.com> | 2021-02-15 15:10:09 +0000 |
commit | 33199c25e5af1553e474a6f6eede07e888cd45ee (patch) | |
tree | 4d7980289479f83eb0cae81a460fb71a184f6dc9 /src/armnn/test | |
parent | 406463269f55a5baefb941b51e10f423f6d3250a (diff) | |
download | armnn-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')
-rw-r--r-- | src/armnn/test/ProfilerTests.cpp | 14 | ||||
-rw-r--r-- | src/armnn/test/UnitTests.hpp | 4 |
2 files changed, 9 insertions, 9 deletions
diff --git a/src/armnn/test/ProfilerTests.cpp b/src/armnn/test/ProfilerTests.cpp index 7bd258e430..a0df3b6b62 100644 --- a/src/armnn/test/ProfilerTests.cpp +++ b/src/armnn/test/ProfilerTests.cpp @@ -19,14 +19,14 @@ namespace armnn { -size_t GetProfilerEventSequenceSize(armnn::Profiler* profiler) +size_t GetProfilerEventSequenceSize(armnn::IProfiler* profiler) { if (!profiler) { return static_cast<size_t>(-1); } - return profiler->m_EventSequence.size(); + return profiler->pProfilerImpl->m_EventSequence.size(); } } // namespace armnn @@ -45,7 +45,7 @@ void RegisterUnregisterProfilerSingleThreadImpl(bool &res) res = !profilerManager.GetProfiler(); // Create and register a profiler for this thread. - std::unique_ptr<armnn::Profiler> profiler = std::make_unique<armnn::Profiler>(); + std::unique_ptr<armnn::IProfiler> profiler = std::make_unique<armnn::IProfiler>(); profilerManager.RegisterProfiler(profiler.get()); // Check that on a single thread we get the same profiler we registered. @@ -66,7 +66,7 @@ BOOST_AUTO_TEST_SUITE(Profiler) BOOST_AUTO_TEST_CASE(EnableDisableProfiling) { - std::unique_ptr<armnn::Profiler> profiler = std::make_unique<armnn::Profiler>(); + std::unique_ptr<armnn::IProfiler> profiler = std::make_unique<armnn::IProfiler>(); // Check that profiling is disabled by default. BOOST_TEST(!profiler->IsProfilingEnabled()); @@ -126,7 +126,7 @@ BOOST_AUTO_TEST_CASE(ProfilingMacros) } // Create and register a profiler for this thread. - std::unique_ptr<armnn::Profiler> profiler = std::make_unique<armnn::Profiler>(); + std::unique_ptr<armnn::IProfiler> profiler = std::make_unique<armnn::IProfiler>(); profilerManager.RegisterProfiler(profiler.get()); { // --- Profiler, but profiling disabled --- @@ -201,7 +201,7 @@ BOOST_AUTO_TEST_CASE(WriteEventResults) armnn::ProfilerManager& profileManager = armnn::ProfilerManager::GetInstance(); // Create and register a profiler for this thread. - std::unique_ptr<armnn::Profiler> profiler = std::make_unique<armnn::Profiler>(); + std::unique_ptr<armnn::IProfiler> profiler = std::make_unique<armnn::IProfiler>(); profileManager.RegisterProfiler(profiler.get()); // Enable profiling. @@ -282,7 +282,7 @@ BOOST_AUTO_TEST_CASE(ProfilerJsonPrinter) armnn::ProfilerManager& profilerManager = armnn::ProfilerManager::GetInstance(); // Create and register a profiler for this thread. - std::unique_ptr<armnn::Profiler> profiler = std::make_unique<armnn::Profiler>(); + std::unique_ptr<armnn::IProfiler> profiler = std::make_unique<armnn::IProfiler>(); profilerManager.RegisterProfiler(profiler.get()); profiler->EnableProfiling(true); 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(); |