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")
  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    "-help")
  31       echo "usage:"
  32       echo "$0 [-np <numprocs>] [-bandwidth <bytes/sec>] [-latency <secs>] program [program-options]"
  33       echo
  34       exit
  35    ;;
  36     *)
  37       break
  38     ;;
  39   esac
  40 done
  41
  42 EXEC="$1"
  43 shift
  44
  45 PLATFORMTMP="$(mktemp tmpXXXXXX)"
  46 #PLATFORMTMP="pla.xml"
  47
  48 cat > ${PLATFORMTMP} <<PLATFORMHEAD
  49 <?xml version='1.0'?>
  50 <!DOCTYPE platform SYSTEM "simgrid.dtd">
  51 <platform version="2">
  52 PLATFORMHEAD
  53
  54 for (( i=${NUMPROCS}; $i ; i=$i-1 )) do
  55   echo "  <host id=\"host$i\" power=\"${POWER}\"/>" >> ${PLATFORMTMP}
  56   echo "  <link id=\"loop$i\" bandwidth=\"${LOOPBACK_BANDWIDTH}\" latency=\"${LOOPBACK_LATENCY}\"/>" >> ${PLATFORMTMP}
  57   echo "  <link id=\"link$i\" bandwidth=\"${NETWORK_BANDWIDTH}\" latency=\"${NETWORK_LATENCY}\"/>" >> ${PLATFORMTMP}
  58 done
  59
  60 for (( i=${NUMPROCS}; $i ; i=$i-1 )) do
  61   for (( j=${NUMPROCS}; $j ; j=$j-1 )) do
  62     if [ $i -eq $j ]; then
  63       echo "  <route src=\"host$i\" dst=\"host$j\"><link:ctn id=\"loop$i\"/></route>" >> ${PLATFORMTMP}
  64     else
  65       echo "  <route src=\"host$i\" dst=\"host$j\"><link:ctn id=\"link$i\"/><link:ctn id=\"link$j\"/></route>" >> ${PLATFORMTMP}
  66     fi
  67   done
  68 done
  69
  70 cat >> ${PLATFORMTMP} <<PLATFORMFOOT
  71 </platform>
  72 PLATFORMFOOT
  73
  74 APPLICATIONTMP="$(mktemp tmpXXXXXX)"
  75 #APPLICATIONTMP="app.xml"
  76
  77 cat > ${APPLICATIONTMP} <<APPLICATIONHEAD
  78 <?xml version='1.0'?>
  79 <!DOCTYPE platform SYSTEM "simgrid.dtd">
  80 <platform version="2">
  81 APPLICATIONHEAD
  82
  83 for (( i=${NUMPROCS}; $i ; i=$i-1 )) do
  84   echo "  <process host=\"host$i\" function=\"smpi_simulated_main\">" >> ${APPLICATIONTMP}
  85   for ARG in $*; do
  86     echo "    <argument value=\"${ARG}\"/>" >> ${APPLICATIONTMP}
  87   done
  88   echo "  </process>" >> ${APPLICATIONTMP}
  89 done
  90
  91 for (( i=${NUMPROCS}; $i ; i=$i-1 )) do
  92   echo "  <process host=\"host$i\" function=\"smpi_sender\"/>" >> ${APPLICATIONTMP}
  93   echo "  <process host=\"host$i\" function=\"smpi_receiver\"/>" >> ${APPLICATIONTMP}
  94 done
  95
  96 cat >> ${APPLICATIONTMP} <<APPLICATIONFOOT
  97 </platform>
  98 APPLICATIONFOOT
  99
 100 ${EXEC} ${PLATFORMTMP} ${APPLICATIONTMP}
 101 rm ${PLATFORMTMP} ${APPLICATIONTMP}