mpi_stress.py

from mpi4py import MPI
import socket
import subprocess
import sys


if __name__ == '__main__':

    threads = sys.argv[1]
    duration = sys.argv[2]

    comm = MPI.COMM_WORLD
    print "Hello! I'm rank %d from %d running on %s" % (comm.rank, comm.size, socket.gethostname())

    subprocess.call(['/ccs/proj/bip149/stress-ng-0.09.34/stress-ng','-c', threads, '-t', duration])

    comm.Barrier() # wait for everybody to synchronize _her