commit 16196e267833178dae62926ff090c50ec6813ad4
author Matthew Bentham <matthew.bentham@arm.com> Mon Apr 01 17:17:58 2019 +0100
committer Matthew Bentham <matthew.bentham@arm.com> Tue Apr 02 10:22:16 2019 +0100
tree 03c0f7cd2490a1e68a63f71ef08c6a6c591f363c
parent a6542c53e929172a36b4d3787686c13a7c569386

MLCE-110 Propagate error from armnn EnqueueWorkload

Change-Id: Ic53b1cdbdd3a7d656932651c74911940affc09b6
Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>

ArmnnDriverImpl.cpp

diff --git a/ArmnnDriverImpl.cpp b/ArmnnDriverImpl.cpp
index a3c2e10..f6456ee 100644
--- a/ArmnnDriverImpl.cpp
+++ b/ArmnnDriverImpl.cpp
@@ -208,7 +208,10 @@
 
     // 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.
-    preparedModel->ExecuteWithDummyInputs();
+    if (!preparedModel->ExecuteWithDummyInputs())
+    {
+        return FailPrepareModel(ErrorStatus::GENERAL_FAILURE, "Network could not be executed", cb);
+    }
 
     if (clTunedParameters &&
         options.GetClTunedParametersMode() == armnn::IGpuAccTunedParameters::Mode::UpdateTunedParameters)