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         shift 2
  33     ;;
  34    "-hostfile")
  35         HOSTFILE="$2"
  36         shift 2
  37     ;;
  38
  39    "-help" | "--help" | "-h")
  40       echo "usage:"
  41       echo "$0 [-np <numprocs>] [-bandwidth <bytes/sec>] [-latency <secs>] [-platform <xmldesc>] [-hostfile <hostfile>] program [program-options]"
  42       echo
  43       exit
  44    ;;
  45     *)
  46       break
  47     ;;
  48   esac
  49 done
  50
  51 EXEC="$1"
  52 shift
  53
  54 ##-------------------------------- DEFAULT or SPECIFIED PLATFORM --------------------------------------
  55 if [ -z "${PLATFORM}" ]; then
  56         PLATFORMTMP="$(mktemp tmpXXXXXX)"
  57         #PLATFORMTMP="pla.xml"
  58
  59         cat > ${PLATFORMTMP} <<PLATFORMHEAD
  60 <?xml version='1.0'?>
  61 <!DOCTYPE platform SYSTEM "simgrid.dtd">
  62 <platform version="2">
  63 PLATFORMHEAD
  64
  65 for (( i=${NUMPROCS}; $i ; i=$i-1 )) do
  66   echo "  <host id=\"host$i\" power=\"${POWER}\"/>" >> ${PLATFORMTMP}
  67   echo "  <link id=\"loop$i\" bandwidth=\"${LOOPBACK_BANDWIDTH}\" latency=\"${LOOPBACK_LATENCY}\"/>" >> ${PLATFORMTMP}
  68   echo "  <link id=\"link$i\" bandwidth=\"${NETWORK_BANDWIDTH}\" latency=\"${NETWORK_LATENCY}\"/>" >> ${PLATFORMTMP}
  69 done
  70
  71 for (( i=${NUMPROCS}; $i ; i=$i-1 )) do
  72   for (( j=${NUMPROCS}; $j ; j=$j-1 )) do
  73     if [ $i -eq $j ]; then
  74       echo "  <route src=\"host$i\" dst=\"host$j\"><link:ctn id=\"loop$i\"/></route>" >> ${PLATFORMTMP}
  75     else
  76       echo "  <route src=\"host$i\" dst=\"host$j\"><link:ctn id=\"link$i\"/><link:ctn id=\"link$j\"/></route>" >> ${PLATFORMTMP}
  77     fi
  78   done
  79 done
  80
  81 cat >> ${PLATFORMTMP} <<PLATFORMFOOT
  82 </platform>
  83 PLATFORMFOOT
  84
  85 else
  86         PLATFORMTMP=${PLATFORM}
  87 fi
  88 ##-------------------------------- end DEFAULT or SPECIFIED PLATFORM --------------------------------------
  89
  90 ##-------------------------------- DEFAULT APPLICATION --------------------------------------
  91 APPLICATIONTMP="$(mktemp tmpXXXXXX)"
  92 #APPLICATIONTMP="app.xml"
  93
  94
  95 cat > ${APPLICATIONTMP} <<APPLICATIONHEAD
  96 <?xml version='1.0'?>
  97 <!DOCTYPE platform SYSTEM "simgrid.dtd">
  98 <platform version="2">
  99 APPLICATIONHEAD
 100
 101 ##---- cache hostnames of hostfile---------------
 102 if [ -n "${HOSTFILE}" ] && [ -f ${HOSTFILE} ]; then
 103         hostnames=(`cat ${HOSTFILE} | tr \\\n " "`)
 104 fi
 105
 106 ##----------------------------------------------------------
 107 ##  generate application.xml with hostnames from hostfile:
 108 ##  the name of host_i (1<=i<=p, where -np p) is the line i
 109 ##  in hostfile (where -hostfile hostfile), or "host$i" if
 110 ##  hostfile has less than i lines.
 111 ##----------------------------------------------------------
 112 for (( i=${NUMPROCS}; $i ; i=$i-1 )) do
 113   j=$(( $i-1 ))
 114   if [ -z "${hostnames[$j]}" ]; then
 115         host="host"$i
 116   else
 117         host="${hostnames[$j]}"
 118   fi
 119   echo "  <process host=\"${host}\" function=\"smpi_simulated_main\">" >> ${APPLICATIONTMP}
 120   for ARG in $*; do
 121     echo "    <argument value=\"${ARG}\"/>" >> ${APPLICATIONTMP}
 122   done
 123   echo "  </process>" >> ${APPLICATIONTMP}
 124 done
 125
 126 for (( i=${NUMPROCS}; $i ; i=$i-1 )) do
 127   j=$(( $i-1 ))
 128   if [ -z "${hostnames[$j]}" ]; then
 129         host="host"$i
 130   else
 131         host="${hostnames[$j]}"
 132   fi
 133
 134   echo "  <process host=\"${host}\" function=\"smpi_sender\"/>" >> ${APPLICATIONTMP}
 135   echo "  <process host=\"${host}\" function=\"smpi_receiver\"/>" >> ${APPLICATIONTMP}
 136 done
 137
 138 cat >> ${APPLICATIONTMP} <<APPLICATIONFOOT
 139 </platform>
 140 APPLICATIONFOOT
 141 ##-------------------------------- end DEFAULT APPLICATION --------------------------------------
 142
 143 ${EXEC} ${PLATFORMTMP} ${APPLICATIONTMP}
 144
 145 echo "[$0] cleaning up temp files"
 146 if [ -z "${PLATFORM}" ]; then
 147         rm ${PLATFORMTMP}
 148 fi
 149 rm ${APPLICATIONTMP}