如何使用 mpi4py 同步内核？

Question

我是运行以下 Python 脚本 mpi4py 版本 3.0.1a0:

from mpi4py import MPI

comm = MPI.COMM_WORLD
rank = comm.Get_rank()
size = comm.Get_size()


print 'INIT',rank,size
comm.Barrier()
if rank==0:
    bla=4
else:
    bla=None
print 'BEFORE',rank,bla
comm.Barrier()
print 'AFTER',rank,bla

并通过 mpiexec -n 16 python test_run.py 提交到集群。似乎一切正常，因为我没有收到任何错误，但是，它没有按照我的预期进行，这意味着它无法识别障碍：

INIT 1 16
BEFORE 1 None
AFTER 1 None
INIT 2 16
BEFORE 2 None
AFTER 2 None
INIT 3 16
BEFORE 3 None
AFTER 3 None
INIT 4 16
BEFORE 4 None
AFTER 4 None
INIT 5 16
BEFORE 5 None
AFTER 5 None
INIT 6 16
BEFORE 6 None
AFTER 6 None
INIT 7 16
BEFORE 7 None
AFTER 7 None
INIT 8 16
BEFORE 8 None
AFTER 8 None
INIT 9 16
BEFORE 9 None
AFTER 9 None
INIT 10 16
BEFORE 10 None
AFTER 10 None
INIT 11 16
BEFORE 11 None
AFTER 11 None
INIT 12 16
BEFORE 12 None
AFTER 12 None
INIT 13 16
BEFORE 13 None
AFTER 13 None
INIT 14 16
BEFORE 14 None
AFTER 14 None
INIT 15 16
BEFORE 15 None
AFTER 15 None
INIT 0 16
BEFORE 0 4
AFTER 0 4

我也尝试添加comm.bcast(bla,root=0)，但这也没有做任何事情。为什么 mpi4py 无法执行这些任务？

Answer 1

问题出在 MPI 上。使用 mopish/3.1.4/intel 而不是 openmpi 并使用此方法重新编译 mpi4py，因此这似乎是一个 platform/infrastructure-dependent 问题。

如何使用 mpi4py 同步内核？

How can I synchronize cores, using mpi4py?

python

parallel-processing

mpi4py