aboutsummaryrefslogtreecommitdiff
path: root/ArmnnDriverImpl.cpp
diff options
context:
space:
mode:
authorFinn Williams <Finn.Williams@arm.com>2021-06-11 15:04:02 +0100
committerFinn Williams <Finn.Williams@arm.com>2021-06-23 13:21:01 +0100
commitca3a3e0fd86a07de9e073ba31dc2b42d6ca84536 (patch)
tree755d270b8c743a24e7ead37dc84f6ca69ad783c4 /ArmnnDriverImpl.cpp
parentdc873f6309784d5fd6914ca5432d32ae6c3de0c2 (diff)
downloadandroid-nn-driver-ca3a3e0fd86a07de9e073ba31dc2b42d6ca84536.tar.gz
IVGCVSW-6062 Rework the async threadpool
!armnn:5801 Signed-off-by: Finn Williams <Finn.Williams@arm.com> Change-Id: I9964d0899ce752441f380edddbd974010257b2dd
Diffstat (limited to 'ArmnnDriverImpl.cpp')
-rw-r--r--ArmnnDriverImpl.cpp6
1 files changed, 3 insertions, 3 deletions
diff --git a/ArmnnDriverImpl.cpp b/ArmnnDriverImpl.cpp
index 0e6e8b18..78ef12fe 100644
--- a/ArmnnDriverImpl.cpp
+++ b/ArmnnDriverImpl.cpp
@@ -166,8 +166,7 @@ Return<V1_0::ErrorStatus> ArmnnDriverImpl<HalPolicy>::prepareModel(
std::string msg;
armnn::INetworkProperties networkProperties(options.isAsyncModelExecutionEnabled(),
armnn::MemorySource::Undefined,
- armnn::MemorySource::Undefined,
- options.getNoOfArmnnThreads());
+ armnn::MemorySource::Undefined);
try
{
@@ -198,7 +197,8 @@ Return<V1_0::ErrorStatus> ArmnnDriverImpl<HalPolicy>::prepareModel(
model,
options.GetRequestInputsAndOutputsDumpDir(),
options.IsGpuProfilingEnabled(),
- options.isAsyncModelExecutionEnabled()));
+ options.isAsyncModelExecutionEnabled(),
+ options.getNoOfArmnnThreads()));
// Run a single 'dummy' inference of the model. This means that CL kernels will get compiled (and tuned if
// this is enabled) before the first 'real' inference which removes the overhead of the first inference.