diff options
author | Finn Williams <Finn.Williams@arm.com> | 2021-06-11 15:04:02 +0100 |
---|---|---|
committer | Finn Williams <Finn.Williams@arm.com> | 2021-06-23 13:21:01 +0100 |
commit | ca3a3e0fd86a07de9e073ba31dc2b42d6ca84536 (patch) | |
tree | 755d270b8c743a24e7ead37dc84f6ca69ad783c4 /ArmnnDriverImpl.cpp | |
parent | dc873f6309784d5fd6914ca5432d32ae6c3de0c2 (diff) | |
download | android-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.cpp | 6 |
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. |