Fixed backend id argument parsing in ExecuteNetwork
authorMatteo Martincigh <matteo.martincigh@arm.com>
Mon, 29 Oct 2018 11:01:09 +0000 (11:01 +0000)
committerLes Bell <les.bell@arm.com>
Mon, 29 Oct 2018 12:45:45 +0000 (12:45 +0000)
Change-Id: I99498a633d75a2b4bdcd059627b07beaa9984890

tests/ExecuteNetwork/ExecuteNetwork.cpp

index 4ebc998..cfddc38 100644 (file)
@@ -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);