src/smpi/smpirun.in

   1 #!/bin/bash
   2 DEFAULT_LOOPBACK_BANDWIDTH="498000000"
   3 DEFAULT_LOOPBACK_LATENCY="0.000004"
   4 DEFAULT_NETWORK_BANDWIDTH="$((26 * 1024 * 1024))"
   5 DEFAULT_NETWORK_LATENCY="0.000005"
   6 DEFAULT_NUMPROCS="4"
   7 DEFAULT_POWER="100"
   8
   9 LOOPBACK_BANDWIDTH="${DEFAULT_LOOPBACK_BANDWIDTH}"
  10 LOOPBACK_LATENCY="${DEFAULT_LOOPBACK_LATENCY}"
  11 NETWORK_BANDWIDTH="${DEFAULT_NETWORK_BANDWIDTH}"
  12 NETWORK_LATENCY="${DEFAULT_NETWORK_LATENCY}"
  13 NUMPROCS="${DEFAULT_NUMPROCS}"
  14 POWER="${DEFAULT_POWER}"
  15
  16 while true; do
  17   case "$1" in
  18    "-np" | "-n")
  19       NUMPROCS="$2"
  20       shift 2
  21     ;;
  22    "-bandwidth")
  23       NETWORK_BANDWIDTH="$2"
  24       shift 2
  25     ;;
  26    "-latency")
  27       NETWORK_LATENCY="$2"
  28       shift 2
  29     ;;
  30    "-platform")
  31         PLATFORM="$2"
  32       if [ ! -f ${PLATFORM} ]; then
  33                 echo "[$0] ** error: the file '${PLATFORM}' does not exist. Aborting."
  34                 exit 1
  35       fi
  36         shift 2
  37     ;;
  38    "-hostfile")
  39         HOSTFILE="$2"
  40       if [ ! -f ${HOSTFILE} ]; then
  41                 echo "[$0] ** error: the file '${HOSTFILE}' does not exist. Aborting."
  42                 exit 1
  43       fi
  44         shift 2
  45     ;;
  46
  47    "-map")
  48         MAPOPT="true"
  49       shift 1
  50    ;;
  51
  52    "-keep-temps")
  53         KEEP="true"
  54       shift 1
  55    ;;
  56
  57    "-help" | "--help" | "-h")
  58       echo "usage:"
  59       echo "$0 [-quiet] [-keep-temps] [-np <numprocs>] -platform <xmldesc> -hostfile <hostfile> [-map] program [program-options]"
  60       echo "or (deprecated usage):"
  61       echo "$0 [-quiet] [-keep-temps] [-np <numprocs>] [-bandwidth <bytes/sec>] [-latency <secs>] program [program-options]"
  62       echo
  63       exit
  64    ;;
  65
  66    "--cfg="*|"--log="*)
  67      for OPT in ${1#*=}
  68      do
  69        SIMOPTS="$SIMOPTS ${1%%=*}=$OPT"
  70      done
  71      shift 1
  72    ;;
  73     *)
  74       break
  75     ;;
  76   esac
  77 done
  78
  79 EXEC="$1"
  80 shift
  81
  82 ##-----------------------------------
  83
  84
  85
  86
  87 ##-------------------------------- DEFAULT or SPECIFIED PLATFORM --------------------------------------
  88 if [ -z "${PLATFORM}" ]; then
  89         PLATFORMTMP="$(mktemp tmpXXXXXX)"
  90
  91         cat > ${PLATFORMTMP} <<PLATFORMHEAD
  92 <?xml version='1.0'?>
  93 <!DOCTYPE platform SYSTEM "simgrid.dtd">
  94 <platform version="2">
  95 PLATFORMHEAD
  96
  97 for (( i=${NUMPROCS}; $i ; i=$i-1 )) do
  98   echo "  <host id=\"host$i\" power=\"${POWER}\"/>" >> ${PLATFORMTMP}
  99   echo "  <link id=\"loop$i\" bandwidth=\"${LOOPBACK_BANDWIDTH}\" latency=\"${LOOPBACK_LATENCY}\"/>" >> ${PLATFORMTMP}
 100   echo "  <link id=\"link$i\" bandwidth=\"${NETWORK_BANDWIDTH}\" latency=\"${NETWORK_LATENCY}\"/>" >> ${PLATFORMTMP}
 101 done
 102
 103 for (( i=${NUMPROCS}; $i ; i=$i-1 )) do
 104   for (( j=${NUMPROCS}; $j ; j=$j-1 )) do
 105     if [ $i -eq $j ]; then
 106       echo "  <route src=\"host$i\" dst=\"host$j\"><link:ctn id=\"loop$i\"/></route>" >> ${PLATFORMTMP}
 107     else
 108       echo "  <route src=\"host$i\" dst=\"host$j\"><link:ctn id=\"link$i\"/><link:ctn id=\"link$j\"/></route>" >> ${PLATFORMTMP}
 109     fi
 110   done
 111 done
 112
 113 cat >> ${PLATFORMTMP} <<PLATFORMFOOT
 114 </platform>
 115 PLATFORMFOOT
 116
 117 else
 118         PLATFORMTMP=${PLATFORM}
 119 fi
 120 ##-------------------------------- end DEFAULT or SPECIFIED PLATFORM --------------------------------------
 121
 122 ##-------------------------------- DEFAULT APPLICATION --------------------------------------
 123 APPLICATIONTMP="$(mktemp tmpXXXXXX)"
 124 #APPLICATIONTMP="app.xml"
 125
 126
 127 cat > ${APPLICATIONTMP} <<APPLICATIONHEAD
 128 <?xml version='1.0'?>
 129 <!DOCTYPE platform SYSTEM "simgrid.dtd">
 130 <platform version="2">
 131 APPLICATIONHEAD
 132
 133 ##---- cache hostnames of hostfile---------------
 134 if [ -n "${HOSTFILE}" ] && [ -f ${HOSTFILE} ]; then
 135         hostnames=(`cat ${HOSTFILE} | tr \\\n " "`)
 136       NUMHOSTS=`cat ${HOSTFILE} | wc -l`
 137 fi
 138
 139 ##----------------------------------------------------------
 140 ##  generate application.xml with hostnames from hostfile:
 141 ##  the name of host_i (1<=i<=p, where -np p) is the line i
 142 ##  in hostfile (where -hostfile hostfile), or "host$i" if
 143 ##  hostfile has less than i lines.
 144 ##----------------------------------------------------------
 145
 146 HAVE_SEQ="`which seq`"
 147 HAVE_JOT="`which jot`"
 148
 149 if [ -n "${HAVE_SEQ}" ]; then
 150         SEQ=`${HAVE_SEQ} 0 $((${NUMPROCS}-1))`
 151 else
 152       cnt=0
 153         while (( $cnt < ${NUMPROCS} )) ; do
 154             SEQ="$SEQ $cnt"
 155             cnt=$((cnt + 1));
 156         done
 157 fi
 158
 159 ##---- generate <process> tags------------------------------
 160
 161 for i in ${SEQ}
 162 do
 163   if [ -n "${HOSTFILE}" ]; then
 164         j=$(( $i % ${NUMHOSTS} ))
 165   fi
 166   ##---- optional display of ranks to process mapping
 167   if [ -n "${MAPOPT}" ]; then
 168         echo "[rank $i] -> ${hostnames[$j]}"
 169   fi
 170
 171   if [ -z "${hostnames[$j]}" ]; then
 172         host="host"$(($j+1))
 173   else
 174         host="${hostnames[$j]}"
 175   fi
 176   echo "  <process host=\"${host}\" function=\"smpi_simulated_main\">" >> ${APPLICATIONTMP}
 177   echo "    <argument value=\"$i\"/> <!-- rank -->" >> ${APPLICATIONTMP}
 178   for ARG in $*; do
 179     echo "    <argument value=\"${ARG}\"/>" >> ${APPLICATIONTMP}
 180   done
 181   echo "  </process>" >> ${APPLICATIONTMP}
 182 done
 183
 184 cat >> ${APPLICATIONTMP} <<APPLICATIONFOOT
 185 </platform>
 186 APPLICATIONFOOT
 187 ##-------------------------------- end DEFAULT APPLICATION --------------------------------------
 188
 189 if [ -n "${KEEP}" ] ; then
 190   echo ${EXEC} ${SIMOPTS} ${PLATFORMTMP} ${APPLICATIONTMP}
 191 fi
 192 ${EXEC} ${SIMOPTS} ${PLATFORMTMP} ${APPLICATIONTMP}
 193
 194 if [ -z "${KEEP}" ] ; then
 195    if [ -z "${PLATFORM}" ]; then
 196         rm ${PLATFORMTMP}
 197    fi
 198    rm ${APPLICATIONTMP}
 199 fi