From ab990de1758905d0b7c7fbae571eb1d8ead5ccda Mon Sep 17 00:00:00 2001
From: ashors1 <ashors@nvidia.com>
Date: Thu, 2 Nov 2023 09:57:11 -0700
Subject: [PATCH] propagate error code to slurm

---
 rosetta/test-vit.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/rosetta/test-vit.sh b/rosetta/test-vit.sh
index 87bc210fa..b74d7e347 100755
--- a/rosetta/test-vit.sh
+++ b/rosetta/test-vit.sh
@@ -136,7 +136,8 @@ with wds.TarWriter(out_tar_path) as dst:
 
 EOF
 
-set -x
+set -exou pipefail
+
 DATA_PATH="/tmp/dummy_vit_data"
 python -m generate_dummy_wds --output_tar_path=${DATA_PATH}
 
@@ -151,5 +152,4 @@ python -m t5x.train \
     --gin_search_paths=/opt/rosetta \
     --gin.CheckpointConfig.save=None \
     $([[ $MULTIPROCESS != 0 ]] && echo --multiprocess_gpu)
-set +x
 echo "Output at ${OUTPUT}"