[ROCm] fix shell bug (microsoft#16641)

`set -ex` with `grep` will exit when grep doesn't meet any string.
shaoboyan091 · Jul 10, 2023 · 2fd5e1c · 2fd5e1c
1 parent 3b729e5
commit 2fd5e1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/ci_build/github/pai/pai_huggingface_bert_large_test.sh b/tools/ci_build/github/pai/pai_huggingface_bert_large_test.sh
@@ -3,7 +3,7 @@
 set -ex
 
 rocm_version=$1
-mi200_gpus=$(rocm-smi --showproductname | grep -c "MI250")
+mi200_gpus=$(rocm-smi --showproductname | grep -c "MI250" | xargs)
 
 echo "mi200_gpus: $mi200_gpus"