diff options
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(); |