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