commit 8fde84b92c211c6499c6ea588886861215835c17
author Finn Williams <Finn.Williams@arm.com> Mon May 31 14:57:15 2021 +0100
committer Finn Williams <Finn.Williams@arm.com> Mon May 31 14:57:15 2021 +0100
tree 416d2fb6a128f4b3e2cb7873521906c63d22ebeb
parent d8fb540568b29fd1d81a1cca667a1ad3e33ef5a1

IVGCVSW-6081 Fix hal 1.0 async execution

Signed-off-by: Finn Williams <Finn.Williams@arm.com>
Change-Id: I4037e3bb47ed0f5585663ca56eed2866bc8a9ef7

ArmnnPreparedModel.cpp

diff --git a/ArmnnPreparedModel.cpp b/ArmnnPreparedModel.cpp
index 978f378..259d1b8 100644
--- a/ArmnnPreparedModel.cpp
+++ b/ArmnnPreparedModel.cpp
@@ -346,7 +346,17 @@
 
     try
     {
-        armnn::Status status = m_Runtime->EnqueueWorkload(m_NetworkId, inputTensors, outputTensors);
+        armnn::Status status;
+        if (m_AsyncModelExecutionEnabled)
+        {
+            ALOGW("ArmnnPreparedModel::ExecuteGraph m_AsyncModelExecutionEnabled true");
+            status = m_Runtime->Execute(*m_WorkingMemHandle, inputTensors, outputTensors);
+        }
+        else
+        {
+            ALOGW("ArmnnPreparedModel::ExecuteGraph m_AsyncModelExecutionEnabled false");
+            status = m_Runtime->EnqueueWorkload(m_NetworkId, inputTensors, outputTensors);
+        }
         if (status != armnn::Status::Success)
         {
             ALOGW("ExecuteWithDummyInputs: EnqueueWorkload failed");