From 6c95836e894f88c4bab6b22f974341f0dd2dddaa Mon Sep 17 00:00:00 2001 From: Cathal Corbett Date: Thu, 20 Oct 2022 16:01:35 +0100 Subject: IVGCVSW-7282 Issues in ExNet when iterations and number of inputs do not match Signed-off-by: Cathal Corbett Change-Id: Ib30fc633a10b6ff8090b50314278fe5dc46fb250 --- tests/ExecuteNetwork/ArmNNExecutor.cpp | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'tests') diff --git a/tests/ExecuteNetwork/ArmNNExecutor.cpp b/tests/ExecuteNetwork/ArmNNExecutor.cpp index 330a239763..fe8443791b 100644 --- a/tests/ExecuteNetwork/ArmNNExecutor.cpp +++ b/tests/ExecuteNetwork/ArmNNExecutor.cpp @@ -304,6 +304,17 @@ void ArmNNExecutor::SetupInputsAndOutputs() " not compatible with number of inputs: " + std::to_string(noOfInputs)); } noInputSets = inputFilePaths / noOfInputs; + // use the same inputSet per iteration or + // there should be a 1:1 mapping between inputSets and iterations + if (noInputSets != 1 || + noInputSets != m_Params.m_Iterations) + { + LogAndThrow("The input model accepts " + std::to_string(noOfInputs) + " inputs. " + + std::to_string(inputFilePaths) + " input-tensor-data file paths have been provided. " + "ExecuteNetwork expects to perform " + std::to_string(noInputSets) + " inference(s) but " + "the iterations number provided is " + std::to_string(m_Params.m_Iterations) + "." + "Please amend the iterations or input-tensor-data parameter."); + } if (noInputSets != 1 && m_Params.m_ReuseBuffers) { LogAndThrow("Specifying multiple sets of inputs not compatible with ReuseBuffers"); -- cgit v1.2.1