在上一篇中我们介绍了 mpi4py 中的 info 和 assertion,下面我们将介绍 mpi4py 中的 Status 对象。
在 MPI-1 中,Status 对象仅用于点到点通信,而 MPI-2 则在通用化请求以及并行 I/O 中使用 Status 对象。因此新的 MPI.Status 对象在不同上下文环境下,可能存在有效域或无效域,特别是在通用化请求操作中,应用程序需要在回调函数中设置某些状态信息。为管理泛化后的 MPI.Status 对象,MPI-2 新增加了下列函数:MPI.Status.Set_elements 和 MPI.Status.Set_cancelled。
需要注意的是,Status 对象最好对应每个操作单独使用,而不推荐复用的方式。此外,如果在通用化请求中需要传递额外信息,则不应修改 Status 对象中各域的语义,而推荐借助 extra_status 参数传递信息。通用化请求将在下一篇中介绍。
下面给出 MPI.Status 对象的方法接口。
MPI.Status.Get_source(self)
获得并返回接收消息的源。也可以通过属性 source 获取。
MPI.Status.Set_source(self, int source)
设置接收消息的源。也可以通过属性 source 设置。
MPI.Status.Get_tag(self)
获得并返回接收消息的 tag。也可以通过属性 tag 获取。
MPI.Status.Set_tag(self, int tag)
设置接收消息的tag。也可以通过属性 tag 设置。
MPI.Status.Get_error(self)
获得接收消息的错误。也可以通过属性 error 获取。
MPI.Status.Set_error(self, int error)
设置接收消息的错误。也可以通过属性 error 设置。
MPI.Status.Get_count(self, Datatype datatype=BYTE)
返回接收到的元素个数(以 datatype
为单位计算)。如果 datatype
的 size 为 0,该方法也会返回 0;如果接收到的数据不是 datatype
的整数倍,则会返回 MPI.UNDEFINED。也可以通过属性 count 获取以 MPI.BYTE 为单位的计数。
MPI.Status.Get_elements(self, Datatype datatype)
返回 datatype
中的基本数据类型的个数。如果支持 MPI-3,mpi4py 实际调用的是 MPI_Get_element