如何使用mpi4py同步内核?

2024-04-20 00:36:04 发布

您现在位置:Python中文网/ 问答频道 /正文

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

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提交给集群。似乎一切都在工作,因为我没有得到任何错误,但是它并没有达到我期望的效果,这意味着它没有识别障碍:

^{pr2}$

我还试图添加comm.bcast(bla,root=0),但这也没有起到任何作用。为什么mpi4py不能执行这些任务?在


Tags: fromimport版本脚本worldsizegetmpi4py