From 3dda41daba5210ebd842115471ecadeb2fb3ae3b Mon Sep 17 00:00:00 2001 From: Jan Eilers Date: Wed, 11 Nov 2020 11:44:14 +0000 Subject: IVGCVSW-5507 ExecuteNetwork fix * Allow multiple backends to be defined seperately like " --compute CpuAcc --compute CpuRef " Signed-off-by: Jan Eilers Change-Id: Ic2a827f6e3463a7413c98a7eefacef8864c1e87b --- tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp b/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp index 72b784c72e..e37c4eb42e 100644 --- a/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp +++ b/tests/ExecuteNetwork/ExecuteNetworkProgramOptions.cpp @@ -161,7 +161,7 @@ ProgramOptions::ProgramOptions() : m_CxxOptions{"ExecuteNetwork", + armnn::BackendRegistryInstance().GetBackendIdsAsString() + " NOTE: Compute devices need to be passed as a comma separated list without whitespaces " "e.g. CpuRef,CpuAcc", - cxxopts::value()) + cxxopts::value>()) ("f,model-format", "armnn-binary, caffe-binary, caffe-text, onnx-binary, onnx-text, tflite-binary, tensorflow-binary or " @@ -352,9 +352,8 @@ void ProgramOptions::ParseOptions(int ac, const char* av[]) CheckOptionDependencies(m_CxxResult); // Some options can't be assigned directly because they need some post-processing: - auto computeDevices = GetOptionValue("compute", m_CxxResult); - m_ExNetParams.m_ComputeDevices = - GetBackendIDs(ParseStringList(computeDevices, ",")); + auto computeDevices = GetOptionValue>("compute", m_CxxResult); + m_ExNetParams.m_ComputeDevices = GetBackendIDs(computeDevices); m_ExNetParams.m_ModelFormat = armnn::stringUtils::StringTrimCopy(GetOptionValue("model-format", m_CxxResult)); m_ExNetParams.m_InputNames = -- cgit v1.2.1