From d63e86c016c7c8049874da3dfc0576d25ddc65fb Mon Sep 17 00:00:00 2001
From: Sadik Armagan <sadik.armagan@arm.com>
Date: Fri, 9 Oct 2020 13:39:17 +0100
Subject: IVGCVSW-5173 'Upgrade from v1.15 of tensorflow to v2.3'

* Fixed the tensorflow build issue.

Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I608c55fdf76ff4d436434c0a50b5368923d2ddb4
---
 scripts/generate_tensorflow_protobuf.sh | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/scripts/generate_tensorflow_protobuf.sh b/scripts/generate_tensorflow_protobuf.sh
index 36fffc0846..5fd7c6b4b1 100755
--- a/scripts/generate_tensorflow_protobuf.sh
+++ b/scripts/generate_tensorflow_protobuf.sh
@@ -59,5 +59,9 @@ find tensorflow -type f -name '*.proto' | grep -v autotuning | grep -v hardware_
     --proto_path=. \
     --proto_path=${PROTOBUF_INSTALL_DIR}/include \
     --cpp_out $OUTPUT_DIR
-  AssertZeroExitCode "Failed to make proto files"
+  EXIT_CODE=$?
+  if [ $EXIT_CODE -ne 0 ]; then
+    echo "Failed to make proto files"
+    exit 1
+  fi
 done
\ No newline at end of file
-- 
cgit v1.2.1