diff options
| author | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-02-11 16:57:38 +0000 |
|---|---|---|
| committer | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-02-12 18:15:12 +0000 |
| commit | cd639c98c347c1addae8c65eb1f4d8e75b468fdb (patch) | |
| tree | d6a4a1ea0d649df94a30272c7247f43071f4e8b1 /ArmnnDriverImpl.cpp | |
| parent | 21469d2498017241d4caac4097479e249d78bf5d (diff) | |
| download | android-nn-driver-cd639c98c347c1addae8c65eb1f4d8e75b468fdb.tar.gz | |
IVGCVSW-5685 Add CpuAcc specific configuration option numberOfThreads
* Added --number-of-threads command line option to android-nn-driver
!armnn:5068
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I248326bf3c0355a7a17b70cb1aac7b7976820c12
Diffstat (limited to 'ArmnnDriverImpl.cpp')
| -rw-r--r-- | ArmnnDriverImpl.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/ArmnnDriverImpl.cpp b/ArmnnDriverImpl.cpp index 917370cf..2381595b 100644 --- a/ArmnnDriverImpl.cpp +++ b/ArmnnDriverImpl.cpp @@ -118,7 +118,8 @@ Return<V1_0::ErrorStatus> ArmnnDriverImpl<HalPolicy>::prepareModel( }); armnn::BackendOptions cpuAcc("CpuAcc", { - { "FastMathEnabled", options.IsFastMathEnabled() } + { "FastMathEnabled", options.IsFastMathEnabled() }, + { "NumberOfThreads", options.GetNumberOfThreads() } }); OptOptions.m_ModelOptions.push_back(gpuAcc); OptOptions.m_ModelOptions.push_back(cpuAcc); |