From 94dd4dbbb2b59f47e38cf2f1280c4b25be45ea64 Mon Sep 17 00:00:00 2001 From: Kevin May Date: Wed, 26 May 2021 16:01:08 +0100 Subject: IVGCVSW-6009 Enable creating thread pool with 1 thread * Allow the user to use create a tread pool with a single thread * This is in keeping with how the android-nn-driver was implemented * Add it to ExecuteNetwork thread pool creation Signed-off-by: Kevin May Change-Id: I05b8048a9e0e45ae11d2b585080af28d9d008d81 --- src/armnn/LoadedNetwork.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/armnn/LoadedNetwork.cpp') diff --git a/src/armnn/LoadedNetwork.cpp b/src/armnn/LoadedNetwork.cpp index 25f8ddf6e5..1a94828841 100644 --- a/src/armnn/LoadedNetwork.cpp +++ b/src/armnn/LoadedNetwork.cpp @@ -170,7 +170,7 @@ LoadedNetwork::LoadedNetwork(std::unique_ptr net, // Create the thread pool which will have working memory handles assigned to each thread // Should occur after factories are registered so that the WorkingMemHandles can be created - if (m_NetworkProperties.m_NumThreads > 1 && networkProperties.m_AsyncEnabled) + if (m_NetworkProperties.m_NumThreads > 0 && networkProperties.m_AsyncEnabled) { CreateThreadPool(m_NetworkProperties.m_NumThreads); } -- cgit v1.2.1