diff options
Diffstat (limited to 'src/armnnTestUtils')
-rw-r--r-- | src/armnnTestUtils/TestUtils.cpp | 6 | ||||
-rw-r--r-- | src/armnnTestUtils/TestUtils.hpp | 4 |
2 files changed, 5 insertions, 5 deletions
diff --git a/src/armnnTestUtils/TestUtils.cpp b/src/armnnTestUtils/TestUtils.cpp index 7c9fc909ac..a67eecfd03 100644 --- a/src/armnnTestUtils/TestUtils.cpp +++ b/src/armnnTestUtils/TestUtils.cpp @@ -56,9 +56,9 @@ ModelOptions& GetModelOptionsForTesting(IOptimizedNetwork* optNet) return optNet->pOptimizedNetworkImpl->GetModelOptions(); } -arm::pipe::ProfilingService& GetProfilingService(armnn::RuntimeImpl* runtime) +arm::pipe::IProfilingService& GetProfilingService(armnn::RuntimeImpl* runtime) { - return runtime->m_ProfilingService; + return *(runtime->m_ProfilingService.get()); } -}
\ No newline at end of file +} diff --git a/src/armnnTestUtils/TestUtils.hpp b/src/armnnTestUtils/TestUtils.hpp index 5433d93332..9fea61b264 100644 --- a/src/armnnTestUtils/TestUtils.hpp +++ b/src/armnnTestUtils/TestUtils.hpp @@ -53,6 +53,6 @@ namespace armnn { Graph& GetGraphForTesting(IOptimizedNetwork* optNetPtr); ModelOptions& GetModelOptionsForTesting(IOptimizedNetwork* optNetPtr); -arm::pipe::ProfilingService& GetProfilingService(RuntimeImpl* runtime); +arm::pipe::IProfilingService& GetProfilingService(RuntimeImpl* runtime); -} // namespace armnn
\ No newline at end of file +} // namespace armnn |