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
diff --git a/source/use_case/kws_asr/src/UseCaseHandler.cc b/source/use_case/kws_asr/src/UseCaseHandler.cc
index 1edc7c4..7025d6d 100644
--- a/source/use_case/kws_asr/src/UseCaseHandler.cc
+++ b/source/use_case/kws_asr/src/UseCaseHandler.cc
@@ -238,7 +238,10 @@
audioDataSlider.TotalStrides() + 1);
/* Run inference over this audio clip sliding window. */
- arm::app::RunInference(kwsModel, profiler);
+ if (!RunInference(kwsModel, profiler)) {
+ printf_err("KWS inference failed\n");
+ return output;
+ }
std::vector<ClassificationResult> kwsClassificationResult;
auto& kwsClassifier = ctx.Get<KwsClassifier&>("kwsclassifier");
@@ -391,7 +394,10 @@
asrPrep.Invoke(asrInferenceWindow, asrInferenceWindowLen, asrInputTensor);
/* Run inference over this audio clip sliding window. */
- arm::app::RunInference(asrModel, profiler);
+ if (!RunInference(asrModel, profiler)) {
+ printf_err("ASR inference failed\n");
+ return false;
+ }
/* Post-process. */
asrPostp.Invoke(asrOutputTensor, reductionAxis, !audioDataSlider.HasNext());