aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xscripts/generate_tensorflow_protobuf.sh6
1 files changed, 5 insertions, 1 deletions
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