| # Copyright 2020 The Microsoft DeepSpeed Team | |
| command -v pdsh | |
| if [ $? != 0 ]; then | |
| echo "Cannot find pdsh, please install via 'apt-get install -y pdsh'" | |
| exit 1 | |
| fi | |
| hostfile=/job/hostfile | |
| while getopts "h?f:" opt; do | |
| case "$opt" in | |
| h|\?) | |
| echo "-f <hostfile>: specify a hostfile, defaults to /job/hostfile" | |
| exit 0 | |
| ;; | |
| f) | |
| hostfile=$OPTARG | |
| shift $((OPTIND-1)) | |
| ;; | |
| esac | |
| done | |
| echo "hostfile=$hostfile" | |
| if [ -f $hostfile ]; then | |
| hosts=`cat $hostfile | awk '{print $1}' | paste -sd "," -` | |
| export PDSH_RCMD_TYPE=ssh | |
| pdsh -w ${hosts} $@ | |
| else | |
| echo "Missing hostfile at ${hostfile}, unable to proceed" | |
| fi | |