aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonBackendModelContext.cpp
diff options
context:
space:
mode:
authorMatthew Sloyan <matthew.sloyan@arm.com>2021-02-10 16:50:53 +0000
committerMatthew Sloyan <matthew.sloyan@arm.com>2021-02-12 17:56:58 +0000
commit0a7dc6bba5d0810fe2ed6f84b0376a8b0674c0b3 (patch)
tree6931067f0e0ff25be7d6a01d379f4b1497dff424 /src/backends/neon/NeonBackendModelContext.cpp
parentb454c5c65efb238c130b042ace390b2bc7f0bf75 (diff)
downloadarmnn-0a7dc6bba5d0810fe2ed6f84b0376a8b0674c0b3.tar.gz
IVGCVSW-5685 Add CpuAcc specific configuration option numberOfThreads
* Added ability to set number of threads used in CpuAcc backend * Enabled number-of-threads option in ExecuteNetwork * Added TfLiteDelegate ModelOptions test * Added unsigned int type to BackendOptions.hpp Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com> Change-Id: Ia576d4f45cbe5df3654bc730bb5ebd5181d82b5a
Diffstat (limited to 'src/backends/neon/NeonBackendModelContext.cpp')
-rw-r--r--src/backends/neon/NeonBackendModelContext.cpp20
1 files changed, 19 insertions, 1 deletions
diff --git a/src/backends/neon/NeonBackendModelContext.cpp b/src/backends/neon/NeonBackendModelContext.cpp
index 2be71e5ded..270592e94d 100644
--- a/src/backends/neon/NeonBackendModelContext.cpp
+++ b/src/backends/neon/NeonBackendModelContext.cpp
@@ -17,13 +17,22 @@ bool ParseBool(const armnn::BackendOptions::Var& value, bool defaultValue)
return defaultValue;
}
+unsigned int ParseUnsignedInt(const armnn::BackendOptions::Var& value, unsigned int defaultValue)
+{
+ if (value.IsUnsignedInt())
+ {
+ return value.AsUnsignedInt();
+ }
+ return defaultValue;
+}
+
} // namespace anonymous
namespace armnn
{
NeonBackendModelContext::NeonBackendModelContext(const ModelOptions& modelOptions)
- : m_IsFastMathEnabled(false)
+ : m_IsFastMathEnabled(false), m_NumberOfThreads(0)
{
if (!modelOptions.empty())
{
@@ -33,6 +42,10 @@ NeonBackendModelContext::NeonBackendModelContext(const ModelOptions& modelOption
{
m_IsFastMathEnabled |= ParseBool(value, false);
}
+ if (name == "NumberOfThreads")
+ {
+ m_NumberOfThreads |= ParseUnsignedInt(value, 0);
+ }
});
}
}
@@ -42,4 +55,9 @@ bool NeonBackendModelContext::IsFastMathEnabled() const
return m_IsFastMathEnabled;
}
+unsigned int NeonBackendModelContext::GetNumberOfThreads() const
+{
+ return m_NumberOfThreads;
+}
+
} // namespace armnn \ No newline at end of file