diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2018-10-23 15:14:46 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-25 09:49:58 +0100 |
commit | 5cc8e56b4ca8d58dc11973c49c10a02a2f13580c (patch) | |
tree | 99f67a0652d2ee27f6cdbdc21c39001e529ee4f1 /tests/MultipleNetworksCifar10 | |
parent | 8853c1f6a802fe7549b89b74ed850aec455b923a (diff) | |
download | armnn-5cc8e56b4ca8d58dc11973c49c10a02a2f13580c.tar.gz |
IVGCVSW-2053: Validate BackendId parameters for the InferenceTest (EndToEnd tests) / ExecuteNetwork
!referencetests:154790
Change-Id: Iaf9608ae1e66460079f2546b30d7fbc9b55bb7dd
Diffstat (limited to 'tests/MultipleNetworksCifar10')
-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)); |