diff options
Diffstat (limited to 'src/profiling/test/ProfilingTestUtils.hpp')
-rw-r--r-- | src/profiling/test/ProfilingTestUtils.hpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/profiling/test/ProfilingTestUtils.hpp b/src/profiling/test/ProfilingTestUtils.hpp index 459d62435b..816ffd3dc6 100644 --- a/src/profiling/test/ProfilingTestUtils.hpp +++ b/src/profiling/test/ProfilingTestUtils.hpp @@ -6,6 +6,7 @@ #pragma once #include "ProfilingUtils.hpp" +#include "Runtime.hpp" #include <armnn/BackendId.hpp> #include <armnn/Optional.hpp> @@ -68,6 +69,11 @@ public: return GetBufferManager(m_ProfilingService); } armnn::profiling::ProfilingService& m_ProfilingService; + + void ForceTransitionToState(ProfilingState newState) + { + TransitionToState(m_ProfilingService, newState); + } }; } // namespace profiling |