动态接收
之前的教程中,传输消息的长度是预先知道的,可以将消息的长度单独的发送/接收,但是 MPI 可通过几个附加的函数调用即可支持动态消息
MPI Status
如果我们将 MPI_Status
传递给 MPI_Recv
函数,那么接收完成后将填充有关接收操作的其他信息。
主要有以下三个信息:
- 发送方的秩,数据被存在
MPI_SOURCE
元素中,如果我们使用MPI_Status stat
,那么秩可这样获得stat.MPI_SOURCE
- 消息的标签,数据被存在
MPI_TAG
元素中 - 消息的长度,没有存在被预定义的元素中,需要使用
MPI_Get_count
获取,返回类型和长度
MPI_Get_count(
MPI_Status* status,
MPI_Datatype datatype,
int* count)
MPI_Recv
可使用 MPI_ANY_SOURCE
和 MPI_ANY_TAG
接收所有秩和标签,这种情况下,MPI_Status
是找出发送方及其标签的唯一结构。不应该相信 MPI_Recv
预订接收的消息长度,如果长度不够会报错。使用 MPI_Get_count
来获得实际可获得的消息长度。
Check-Status (查询 MPI 状态程序)
#include <mpi.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
int main() {
MPI_Init(NULL, NULL);
int world_size;
MPI_Comm_size(MPI_COMM_WORLD, &world_size);
if (world_size != 2) {
fprintf(stderr, "Must use two processes fo