diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 4 | ||||
| -rw-r--r-- | tests/profiling/gatordmock/tests/GatordMockTests.cpp | 2 |
2 files changed, 4 insertions, 2 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index abcda40d1d..1f19584c68 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -11,6 +11,7 @@ #include <armnn/Logging.hpp> #include <armnnUtils/Filesystem.hpp> #include <armnnUtils/TContainer.hpp> +#include <ProfilingOptionsConverter.hpp> #include <InferenceTest.hpp> #if defined(ARMNN_SERIALIZER) @@ -90,7 +91,8 @@ int TfLiteDelegateMainImpl(const ExecuteNetworkParams& params, const armnn::IRun // Create the Armnn Delegate // Populate a DelegateOptions from the ExecuteNetworkParams. armnnDelegate::DelegateOptions delegateOptions = params.ToDelegateOptions(); - delegateOptions.SetExternalProfilingParams(runtimeOptions.m_ProfilingOptions); + delegateOptions.SetExternalProfilingParams( + ConvertExternalProfilingOptions(runtimeOptions.m_ProfilingOptions)); std::unique_ptr<TfLiteDelegate, decltype(&armnnDelegate::TfLiteArmnnDelegateDelete)> theArmnnDelegate(armnnDelegate::TfLiteArmnnDelegateCreate(delegateOptions), diff --git a/tests/profiling/gatordmock/tests/GatordMockTests.cpp b/tests/profiling/gatordmock/tests/GatordMockTests.cpp index 388de368dd..f76b2939cd 100644 --- a/tests/profiling/gatordmock/tests/GatordMockTests.cpp +++ b/tests/profiling/gatordmock/tests/GatordMockTests.cpp @@ -250,7 +250,7 @@ TEST_CASE("GatorDMockEndToEnd") arm::pipe::ConnectionHandler connectionHandler(udsNamespace, false); // Enable the profiling service. - armnn::IRuntime::CreationOptions::ExternalProfilingOptions options; + armnn::profiling::ProfilingOptions options; options.m_EnableProfiling = true; options.m_TimelineEnabled = true; |