mpi - 如何访问发送到 MPI 中处理器的最后一条消息？

Question

我使用的是使用消息传递接口(interface)(MPI)的主从结构，但是每当我调用接收函数时，我需要接收从主节点发送到主节点的最后一条消息，而不是按照发送顺序接收消息。每个处理器并忽略前面的处理器!

我的问题是，是否有任何方法可以访问每个处理器的缓冲区并选择队列中的最后一条消息？

Answer 1

不，你不能只是查看队列；但您可以测试是否存在更多消息 MPI_Probe或MPI_Iprobe ，当存在更多消息时，继续接收并丢弃旧数据:

#!/usr/bin/env python
from mpi4py import MPI
import time

def waiter(comm, sendTask):
    # wait for messages to be present
    while not comm.Iprobe(source=sendTask, tag=1):
        time.sleep(1)

    # read all messages while more are available, discarding old
    while comm.Iprobe(source=sendTask, tag=1):
        lastMsg = comm.recv(source=sendTask, tag=1)

    if lastMsg is None:
        print "No messages pending"
    else:
        print "Last message was ", lastMsg

    comm.Barrier() 

def sender(comm, waitTask):
    for msgno in range(5):
        print "sending: ", msgno
        comm.send(msgno, dest=waitTask, tag=1)

    print "sending: ", -1
    comm.send(-1, dest=waitTask, tag=1)

    comm.Barrier() 

if __name__== "__main__":
    comm = MPI.COMM_WORLD
    sendTask = 1
    waitTask = 0
    if comm.rank == waitTask:
        waiter(comm, sendTask)
    elif comm.rank == sendTask:
        sender(comm, waitTask)
    else:
        comm.Barrier()

运行给予

$ mpirun -np 2 ./readall.py
sending:  0
sending:  1
sending:  2
sending:  3
sending:  4
sending:  -1
Last message was  -1