diff options
author | Jim Flynn <jim.flynn@arm.com> | 2022-03-04 15:03:58 +0000 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2022-03-06 19:12:00 +0000 |
commit | 344302581b66677a748a456f370752db75adde21 (patch) | |
tree | 6463b11c70c501e46a2a93d7d57f4f68e4b78cb6 /src/armnn/test/RuntimeTests.cpp | |
parent | 9c315305aaa31a3d1610738cd40f558381f8c687 (diff) | |
download | armnn-344302581b66677a748a456f370752db75adde21.tar.gz |
IVGCVSW-6816 Inject counter registration into ProfilingService
Change-Id: I87ce3a1306eced9fc347cc383d9c7bc8994f0b0c
Signed-off-by: Jim Flynn <jim.flynn@arm.com>
Diffstat (limited to 'src/armnn/test/RuntimeTests.cpp')
-rw-r--r-- | src/armnn/test/RuntimeTests.cpp | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/src/armnn/test/RuntimeTests.cpp b/src/armnn/test/RuntimeTests.cpp index 73e36eaf56..afe4bc7ea0 100644 --- a/src/armnn/test/RuntimeTests.cpp +++ b/src/armnn/test/RuntimeTests.cpp @@ -6,6 +6,8 @@ #include <armnn/Descriptors.hpp> #include <armnn/IRuntime.hpp> #include <armnn/INetwork.hpp> +#include <armnn/profiling/ArmNNProfiling.hpp> +#include <ArmNNProfilingServiceInitialiser.hpp> #include <ProfilingOptionsConverter.hpp> #include <Processes.hpp> #include <Runtime.hpp> @@ -627,7 +629,9 @@ TEST_CASE("ProfilingDisable") armnn::NetworkId netId; CHECK(runtime.LoadNetwork(netId, std::move(optNet)) == Status::Success); - ProfilingServiceRuntimeHelper profilingServiceHelper(GetProfilingService(&runtime)); + armnn::ArmNNProfilingServiceInitialiser initialiser; + ProfilingServiceRuntimeHelper profilingServiceHelper( + arm::pipe::MAX_ARMNN_COUNTER, initialiser, GetProfilingService(&runtime)); BufferManager& bufferManager = profilingServiceHelper.GetProfilingBufferManager(); auto readableBuffer = bufferManager.GetReadableBuffer(); @@ -649,7 +653,9 @@ TEST_CASE("ProfilingEnableCpuRef") GetProfilingService(&runtime).ResetExternalProfilingOptions( ConvertExternalProfilingOptions(options.m_ProfilingOptions), false); - ProfilingServiceRuntimeHelper profilingServiceHelper(GetProfilingService(&runtime)); + armnn::ArmNNProfilingServiceInitialiser initialiser; + ProfilingServiceRuntimeHelper profilingServiceHelper( + arm::pipe::MAX_ARMNN_COUNTER, initialiser, GetProfilingService(&runtime)); profilingServiceHelper.ForceTransitionToState(ProfilingState::NotConnected); profilingServiceHelper.ForceTransitionToState(ProfilingState::WaitingForAck); profilingServiceHelper.ForceTransitionToState(ProfilingState::Active); |