mpi4py 点到点通信之缓冲阻塞通信模式

本文从本人简书博客同步过来

在上一篇中我们介绍了 mpi4py 中标准阻塞通信模式，下面我们将介绍缓冲阻塞通信模式。

缓冲通信模式主要用于解开阻塞通信的发送和接收之间的耦合。有了缓冲机制，即使在接受端没有启动相应的接收的情况下，在完成其消息数据到缓冲区的转移后发送端的阻塞发送函数也可返回。其实标准通信模式中也存在缓冲机制，它使用的是 MPI 环境所提供的数据缓冲区，是有一定大小的。使用缓冲通信模式，我们可以自己分配和组装一块内存区域用作缓冲区，缓冲区的大小可以根据需要进行控制。但需要注意的是，当消息大小超过缓冲区容量时，程序会出错。

下面是 mpi4py 中用于缓冲阻塞点到点通信的方法接口（MPI.Comm 类的方法）：

bsend(self, obj, int dest, int tag=0)
recv(self, buf=None, int source=ANY_SOURCE, int tag=ANY_TAG, Status status=None)

Bsend(self, buf, int dest, int tag=0)
Recv(self, buf, int source=ANY_SOURCE, int tag=ANY_TAG, Status status=None)

这些方法调用中的参数是与标准通信模式的方法调用参数一样的。

另外我们会用到的装配和卸载用于通信的缓冲区的函数如下：

MPI.Attach_buffer(buf)
MPI.Detach_buffer()

下面分别给出 bsend/recv 和 Bsend/Recv 的使用例程。

# bsend_recv.py

from mpi4py import MPI


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

# MPI.BSEND_OVERHEAD gives the extra overhead in buffered mode
BUFSISE = 2000 + MPI.BSEND_OVERHEAD
buf = bytearray(BUFSISE)

# Attach a user-provided buffer for sending in buffered mode
MPI.Attach_buffer(buf)

send_obj = {
  'a': [1, 2.4, 'abc', -2.3+3.4J],
            'b': {
  2, 3, 4}}

if rank == 0:
    comm.bsend(send_obj, dest=1, tag=