diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-01-31 16:44:26 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-02-01 09:10:51 +0000 |
commit | 339bcae73515c66899432b5844d7c239c570c4b8 (patch) | |
tree | 91bea3fc1eda5cf96309ad2a255917667ffad679 /tests/ExecuteNetwork/ExecuteNetwork.cpp | |
parent | 0c051f9b6b8d7d1602e81d2977dda449b6392642 (diff) | |
download | armnn-339bcae73515c66899432b5844d7c239c570c4b8.tar.gz |
IVGCVSW-2604 Fix bug that made it impossible to execute inference tests on certain backends
* Read compute devices from the CL as strings and convert them into BackendId objects afterwards
Change-Id: Icded1c572778f5a213644e3052ff6dfe7022128b
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetwork.cpp')
-rw-r--r-- | tests/ExecuteNetwork/ExecuteNetwork.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp index 29780104c2..afde9860e2 100644 --- a/tests/ExecuteNetwork/ExecuteNetwork.cpp +++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp @@ -171,7 +171,7 @@ void RemoveDuplicateDevices(std::vector<armnn::BackendId>& computeDevices) template<typename TParser, typename TDataType> int MainImpl(const char* modelPath, bool isModelBinary, - const std::vector<armnn::BackendId>& computeDevice, + const std::vector<armnn::BackendId>& computeDevices, const char* inputName, const armnn::TensorShape* inputTensorShape, const char* inputTensorDataFilePath, @@ -200,7 +200,7 @@ int MainImpl(const char* modelPath, typename InferenceModel<TParser, TDataType>::Params params; params.m_ModelPath = modelPath; params.m_IsModelBinary = isModelBinary; - params.m_ComputeDevice = computeDevice; + params.m_ComputeDevices = computeDevices; params.m_InputBindings = { inputName }; params.m_InputShapes = { *inputTensorShape }; params.m_OutputBindings = { outputName }; |