From 14e400226ad50a262b388f256d8bfd619ee09a7b Mon Sep 17 00:00:00 2001
From: nachocano <nachoacano@gmail.com>
Date: Sat, 6 Dec 2014 11:33:05 -0800
Subject: [PATCH] submit mpi to include machine file

---
 submit_mpi.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/submit_mpi.py b/submit_mpi.py
index 14eafff8c..83c97175e 100755
--- a/submit_mpi.py
+++ b/submit_mpi.py
@@ -24,13 +24,13 @@ def mpi_submit(nslave, args):
          args arguments to launch each job
               this usually includes the parameters of master_uri and parameters passed into submit
     """
-    cmd = ' '.join(['mpirun -n %d' % nslave] + args)
+    cmd = ' '.join(['mpirun -n %d -machinefile %s' % (nslave, args[0])] + args[1:])
     print cmd
     subprocess.check_call(cmd, shell = True)
 
 if __name__ == '__main__':
     if len(sys.argv) < 2:
-        print 'Usage: <nslave> <cmd>'
+        print 'Usage: <nslave> <machine_file> <cmd>'
         exit(0)        
     # call submit, with nslave, the commands to run each job and submit function
     tracker.submit(int(sys.argv[1]), sys.argv[2:], fun_submit= mpi_submit)