aboutsummaryrefslogtreecommitdiff
path: root/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp
diff options
context:
space:
mode:
authorDavid Beck <david.beck@arm.com>2018-10-19 15:20:56 +0100
committerMatthew Bentham <matthew.bentham@arm.com>2018-10-22 16:57:54 +0100
commitf0b4845c1c6f24f59d4c88473b852cf69a3c7ae9 (patch)
tree5a8726ee4a397c421a6a41d6edca1a2d3183f168 /tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp
parent7bc8c9fc9726d3c9ac002138c594688a006faac6 (diff)
downloadarmnn-f0b4845c1c6f24f59d4c88473b852cf69a3c7ae9.tar.gz
IVGCVSW-2019 : replace Compute enum in the backend preferences list
Change-Id: Ie7549fd27378acfa97e68d098e338b8c9d4ea5d2
Diffstat (limited to 'tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp')
-rw-r--r--tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp8
1 files changed, 4 insertions, 4 deletions
diff --git a/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp b/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp
index 34fdbf0867..f9fdf8b3ea 100644
--- a/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp
+++ b/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp
@@ -36,7 +36,8 @@ int main(int argc, char* argv[])
namespace po = boost::program_options;
- std::vector<armnn::Compute> computeDevice;
+ std::vector<armnn::BackendId> computeDevice;
+ std::vector<armnn::BackendId> defaultBackends = {armnn::Compute::CpuAcc, armnn::Compute::CpuRef};
std::string modelDir;
std::string dataDir;
@@ -48,8 +49,7 @@ int main(int argc, char* argv[])
("help", "Display help messages")
("model-dir,m", po::value<std::string>(&modelDir)->required(),
"Path to directory containing the Cifar10 model file")
- ("compute,c", po::value<std::vector<armnn::Compute>>(&computeDevice)->default_value
- ({armnn::Compute::CpuAcc, armnn::Compute::CpuRef}),
+ ("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")
("data-dir,d", po::value<std::string>(&dataDir)->required(),
"Path to directory containing the Cifar10 test data");
@@ -200,7 +200,7 @@ int main(int argc, char* argv[])
}
catch (const std::exception& e)
{
- // Coverity fix: various boost exceptions can be thrown by methods called by this test.
+ // Coverity fix: various boost exceptions can be thrown by methods called by this test.
std::cerr << "WARNING: MultipleNetworksCifar10: An error has occurred when running the "
"multiple networks inference tests: " << e.what() << std::endl;
return 1;