使用mpi4py时如何只运行一次

Question

假设我有一个简单的程序，foo.py：

from mpi4py import MPI
comm = MPI.COMM_WORLD
rank = comm.Get_rank()

# Some stuff happens here

print "Done"

并使用mpiexec -n 4 python foo.py运行它，打印“完成”4次。

如何在结束时只打印一次“完成”？

我试过

if rank == 3:
    print "Done"

但这不能保证是最后的。有关mpi4py的I / O的文档不完整。