diff options
author | David Beck <david.beck@arm.com> | 2018-10-19 15:20:56 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-10-22 16:57:54 +0100 |
commit | f0b4845c1c6f24f59d4c88473b852cf69a3c7ae9 (patch) | |
tree | 5a8726ee4a397c421a6a41d6edca1a2d3183f168 /tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp | |
parent | 7bc8c9fc9726d3c9ac002138c594688a006faac6 (diff) | |
download | armnn-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.cpp | 8 |
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; |