Question

我有一个进程正在运行一个名为t1.py的程序，它产生了3个其他进程，所有进程都运行t2.py.我想从生成的进程中广播一个值为0的值到其他两个生成的进程。但是，当调用bcast时，程序会阻塞。知道为什么会这样吗？我该如何解决？

t1.py

from mpi4py import MPI
import sys

sub_comm = MPI.COMM_SELF.Spawn(sys.executable, args=['t2.py'], maxprocs=3)
print 'hi'

t2.py

from mpi4py import MPI

comm = MPI.Comm.Get_Parent()

print 'ho ', comm.Get_rank()
a = comm.bcast(comm.Get_rank(), root=0)
print a

输出

hi
ho  2
ho  0
ho  1

Answer 1

如果您只是希望孩子互相交谈，可以使用MPI.COMM_WORLD：

a = MPI.COMM_WORLD.bcast(MPI.COMM_WORLD.Get_rank(), root=0)

通过打印MPI.COMM_WORLD.Get_rank(), ' of ',MPI.COMM_WORLD.Get_size()，您可以检查孩子＆＃39; MPI.COMM_WORLD是否仅限于孩子。

现在，让我们调查comm.bcast(...)在comm获得comm=MPI.Comm.Get_parent()后失败的原因。实际上，通过查看此通信器的大小和等级，它似乎与MPI.COMM_WORLD非常相似。 但相反，comm与MPI.COMM_WORLD非常不同：它是intercommunicator 。更确切地说，这是父母可以与孩子交谈的方式。可以使用集体通信，但所有进程（父进程及其子进程）都必须调用该函数。请仔细阅读MPI standards，特别是关于Intercommunicator Collective Operations的第5.2.2和5.2.3节。关于bcast()，使用MPI.ROOT和MPI.PROC_NULL代替广播公司root的排名来指定方向（子级到子级的父级）以及发送过程。最后，可以使用Merge()（对应于MPI_Intercomm_merge()）在互通者的基础上定义一个内部通信器。在这个内部通信器中，父母和孩子不属于两个不同的群体：他们是像往常一样以其独特等级为特征的过程。

以下是t1.py和t2.py的修改版本，其中执行了一个bcast()的内部通信器。然后，相互通信器为Merge()，并且通常会调用生成的内部通信器上的bcast()。

t1.py

from mpi4py import MPI
import sys

sub_comm = MPI.COMM_SELF.Spawn(sys.executable, args=['t2.py'], maxprocs=3)

val=42
sub_comm.bcast(val, MPI.ROOT)

common_comm=sub_comm.Merge(False)
print 'parent in common_comm ', common_comm.Get_rank(), ' of  ',common_comm.Get_size()
#MPI_Intercomm_merge(parentcomm,1,&intracomm);

val=13
c=common_comm.bcast(val, root=0)
print "value from rank 0 in common_comm", c

t2.py

from mpi4py import MPI

comm = MPI.Comm.Get_parent()

print 'ho ', comm.Get_rank(), ' of  ',comm.Get_size(),' ', MPI.COMM_WORLD.Get_rank(), ' of  ',MPI.COMM_WORLD.Get_size()
a = MPI.COMM_WORLD.bcast(MPI.COMM_WORLD.Get_rank(), root=0)
print "value from other child", a

print "comm.Is_inter", comm.Is_inter()
b = comm.bcast(comm.Get_rank(), root=0)
print "value from parent", b

common_comm=comm.Merge(True)
print "common_comm.Is_inter", common_comm.Is_inter()
print 'common_comm ', common_comm.Get_rank(), ' of  ',common_comm.Get_size()

c=common_comm.bcast(0, root=0)
print "value from rank 0 in common_comm", c

mpi4py：在生成的进程之间进行通信

1 个答案: