aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMatteo Martincigh <matteo.martincigh@arm.com>2018-10-29 11:01:09 +0000
committerLes Bell <les.bell@arm.com>2018-10-29 12:45:45 +0000
commit067112faac5aeb1d28d698115c4c2d5b051e7df1 (patch)
treedfe8b9abe3d5cb99fd6bd72fe4997571b88cd10b
parent7af7688a77e9fe2929a626f7c7dab9aef089d2b3 (diff)
downloadarmnn-067112faac5aeb1d28d698115c4c2d5b051e7df1.tar.gz
Fixed backend id argument parsing in ExecuteNetwork
Change-Id: I99498a633d75a2b4bdcd059627b07beaa9984890
-rw-r--r--tests/ExecuteNetwork/ExecuteNetwork.cpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp
index 4ebc9987a5..cfddc38a99 100644
--- a/tests/ExecuteNetwork/ExecuteNetwork.cpp
+++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp
@@ -589,7 +589,8 @@ int main(int argc, const char* argv[])
else // Run single test
{
// Get the preferred order of compute devices.
- std::vector<armnn::BackendId> computeDevices = vm["compute"].as<std::vector<armnn::BackendId>>();
+ std::vector<std::string> computeDevicesAsStrings = vm["compute"].as<std::vector<std::string>>();
+ std::vector<armnn::BackendId> computeDevices(computeDevicesAsStrings.begin(), computeDevicesAsStrings.end());
// Remove duplicates from the list of compute devices.
RemoveDuplicateDevices(computeDevices);