diff options
Diffstat (limited to 'tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp')
| -rw-r--r-- | tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp b/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp index f9fdf8b3ea..f31e0c95a9 100644 --- a/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp +++ b/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp @@ -41,6 +41,9 @@ int main(int argc, char* argv[]) std::string modelDir; std::string dataDir; + const std::string backendsMessage = "Which device to run layers on by default. Possible choices: " + + armnn::BackendRegistryInstance().GetBackendIdsAsString(); + po::options_description desc("Options"); try { @@ -50,7 +53,7 @@ int main(int argc, char* argv[]) ("model-dir,m", po::value<std::string>(&modelDir)->required(), "Path to directory containing the Cifar10 model file") ("compute,c", po::value<std::vector<armnn::BackendId>>(&computeDevice)->default_value(defaultBackends), - "Which device to run layers on by default. Possible choices: CpuAcc, CpuRef, GpuAcc") + backendsMessage.c_str()) ("data-dir,d", po::value<std::string>(&dataDir)->required(), "Path to directory containing the Cifar10 test data"); } @@ -91,6 +94,15 @@ int main(int argc, char* argv[]) } string modelPath = modelDir + "cifar10_full_iter_60000.caffemodel"; + // Check if the requested backend are all valid + std::string invalidBackends; + if (!CheckRequestedBackendsAreValid(computeDevice, armnn::Optional<std::string&>(invalidBackends))) + { + BOOST_LOG_TRIVIAL(fatal) << "The list of preferred devices contains invalid backend IDs: " + << invalidBackends; + return EXIT_FAILURE; + } + // Create runtime armnn::IRuntime::CreationOptions options; armnn::IRuntimePtr runtime(armnn::IRuntime::Create(options)); |