commit b254731ff27a40f382695d5753e1b537c4736bfa
author David Monahan <David.Monahan@arm.com> Thu Feb 24 15:55:56 2022 +0000
committer Colm Donelan <colm.donelan@arm.com> Fri Feb 25 10:09:10 2022 +0000
tree cb504a2504a03a15e097583005ca54386f12ba05
parent f4809b5f8c166b8b89ced34a9809fc7f4d4b70bf

IVGCVSW-6801 Remove the call to sync after EnqueueWorkload if we didn't import

Signed-off-by: David Monahan <David.Monahan@arm.com>
Change-Id: I0c79518fb0d849809d972241ce02653259d4f8c4

src/armnn/Runtime.cpp

diff --git a/src/armnn/Runtime.cpp b/src/armnn/Runtime.cpp
index 1abe0f3..a913681 100644
--- a/src/armnn/Runtime.cpp
+++ b/src/armnn/Runtime.cpp
@@ -630,15 +630,18 @@
     auto status = loadedNetwork->EnqueueWorkload(inputTensors, outputTensors,
                                                  preImportedInputIds, preImportedOutputIds);
 
+
+    // Check if we imported, if not there's no need to call the After EnqueueWorkload events
+    if (!preImportedInputIds.empty() || !preImportedOutputIds.empty())
+    {
+        // Call After EnqueueWorkload events
+        for (auto&& context : m_BackendContexts)
+        {
+            context.second->AfterEnqueueWorkload(networkId);
+        }
+    }
     ARMNN_LOG(info) << "Execution time: " << std::setprecision(2)
                     << std::fixed << armnn::GetTimeDuration(startTime).count() << " ms.";
-
-    // Call After EnqueueWorkload events
-    for (auto&& context : m_BackendContexts)
-    {
-        context.second->AfterEnqueueWorkload(networkId);
-    }
-
     return status;
 }