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: ' + print 'Usage: ' 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)