From 27b62d9dec86b76422c6c8be7615c1f712518570 Mon Sep 17 00:00:00 2001
From: alexander <alexander.efremov@arm.com>
Date: Tue, 4 May 2021 20:46:08 +0100
Subject: If inference fails - exit before printing results. It was easy to
 miss error message and mistakenly assume that execution succeeded.

Signed-off-by: alexander <alexander.efremov@arm.com>
Change-Id: I3c7beb5afe97db5df486e12e3bfe0e8e27e382f8
---
 source/use_case/inference_runner/src/UseCaseHandler.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'source/use_case/inference_runner/src')

diff --git a/source/use_case/inference_runner/src/UseCaseHandler.cc b/source/use_case/inference_runner/src/UseCaseHandler.cc
index a2b3fb7..a75b2e4 100644
--- a/source/use_case/inference_runner/src/UseCaseHandler.cc
+++ b/source/use_case/inference_runner/src/UseCaseHandler.cc
@@ -68,7 +68,9 @@ namespace app {
                                 str_inf.c_str(), str_inf.size(),
                                 dataPsnTxtInfStartX, dataPsnTxtInfStartY, 0);
 
-        RunInference(model, profiler);
+        if (!RunInference(model, profiler)) {
+            return false;
+        }
 
         /* Erase. */
         str_inf = std::string(str_inf.size(), ' ');
-- 
cgit v1.2.1