aboutsummaryrefslogtreecommitdiff
path: root/tests/ExecuteNetwork/ExecuteNetwork.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/ExecuteNetwork/ExecuteNetwork.cpp')
-rw-r--r--tests/ExecuteNetwork/ExecuteNetwork.cpp7
1 files changed, 5 insertions, 2 deletions
diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp
index 7d6aafcfbf..dd769755b4 100644
--- a/tests/ExecuteNetwork/ExecuteNetwork.cpp
+++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp
@@ -588,8 +588,11 @@ int main(int argc, const char* argv[])
}
else // Run single test
{
- // Get the preferred order of compute devices.
- std::vector<std::string> computeDevicesAsStrings = vm["compute"].as<std::vector<std::string>>();
+ // Get the preferred order of compute devices. If none are specified, default to using CpuRef
+ const std::string computeOption("compute");
+ std::vector<std::string> computeDevicesAsStrings = CheckOption(vm, computeOption.c_str()) ?
+ vm[computeOption].as<std::vector<std::string>>() :
+ std::vector<std::string>({ "CpuRef" });
std::vector<armnn::BackendId> computeDevices(computeDevicesAsStrings.begin(), computeDevicesAsStrings.end());
// Remove duplicates from the list of compute devices.