From 19a1c0347709506c5ad5c1b7f647a5af4a30fba0 Mon Sep 17 00:00:00 2001
From: Sadik Armagan <sadik.armagan@arm.com>
Date: Wed, 20 Jan 2021 12:17:00 +0000
Subject: MLCE-336 'TfLite parser and ExecuteNetwork issues'

* Report the error thrown

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: If5e5160f43c11a284b919d76b04d9c4d53e9c56c
---
 tests/ExecuteNetwork/ExecuteNetwork.cpp | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'tests')

diff --git a/tests/ExecuteNetwork/ExecuteNetwork.cpp b/tests/ExecuteNetwork/ExecuteNetwork.cpp
index 6880a17c52..5df5dfbce7 100644
--- a/tests/ExecuteNetwork/ExecuteNetwork.cpp
+++ b/tests/ExecuteNetwork/ExecuteNetwork.cpp
@@ -61,6 +61,11 @@ int TfLiteDelegateMainImpl(const ExecuteNetworkParams& params,
                              armnnDelegate::TfLiteArmnnDelegateDelete);
     // Register armnn_delegate to TfLiteInterpreter
     int status = tfLiteInterpreter->ModifyGraphWithDelegate(std::move(theArmnnDelegate));
+    if (status == kTfLiteError)
+    {
+        ARMNN_LOG(fatal) << "Could not register ArmNN TfLite Delegate to TfLiteInterpreter!";
+        return EXIT_FAILURE;
+    }
 
     std::vector<std::string>  inputBindings;
     for (const std::string& inputName: params.m_InputNames)
-- 
cgit v1.2.1