MPI集体通信替代循环接收：MPI_Gather使用问题求助

错误分析 #

你当前代码的核心问题是MPI_Gather的参数不匹配，导致运行异常：

• 根进程的recvcount参数错误：你写的total_rows + 1应该改为total_rows，这个参数代表每个发送进程传递的元素数量，必须和其他进程的sendcount保持一致（若需处理不同长度的片段，需改用MPI_Gatherv）。
• 根进程接收缓冲区c的大小N必须等于进程总数 × total_rows，否则会出现内存不足或数据截断问题。

正确实现代码 #

假设总进程数为size（可通过MPI_Comm_size(MPI_COMM_WORLD, &size)获取），每个非根进程持有total_rows个double元素的result数组，根进程需收集所有片段拼成完整向量：

根进程（rank=0）代码 #

int size;
MPI_Comm_size(MPI_COMM_WORLD, &size);
// 分配足够容纳所有进程数据的缓冲区
double *c = malloc(size * total_rows * sizeof(double));
if (c == NULL) {
    MPI_Abort(MPI_COMM_WORLD, 1);
}

// MPI_Gather参数说明：
// sendbuf: 根进程无需发送数据，填NULL
// sendcount: 根进程发送元素数，填0
// sendtype: 对应sendbuf的类型，填MPI_DOUBLE
// recvbuf: 接收所有进程数据的缓冲区
// recvcount: 每个进程发送的元素数，即total_rows
// recvtype: 接收元素类型，MPI_DOUBLE
// root: 根进程rank，这里为0
// comm: 通信域，MPI_COMM_WORLD
MPI_Gather(NULL, 0, MPI_DOUBLE,
           c, total_rows, MPI_DOUBLE, 0, MPI_COMM_WORLD);

// 使用完毕后释放内存
free(c);

其他进程（rank≠0）代码 #

// MPI_Gather参数说明：
// sendbuf: 本地待发送的数组，即result
// sendcount: 发送的元素数量，即total_rows
// sendtype: 发送元素类型，MPI_DOUBLE
// recvbuf: 非根进程无需接收数据，填NULL
// recvcount: 非根进程该参数无实际意义，通常与根进程保持一致即可
// recvtype: 对应recvbuf的类型，填MPI_DOUBLE
// root: 根进程rank，0
// comm: 通信域，MPI_COMM_WORLD
MPI_Gather(result, total_rows, MPI_DOUBLE,
           NULL, total_rows, MPI_DOUBLE, 0, MPI_COMM_WORLD);

关键注意事项 #

• 所有进程必须调用MPI_Gather：集体通信要求参与通信的所有进程都执行该调用，根进程也不能例外。
• 参数一致性：非根进程的sendcount必须和根进程的recvcount匹配，确保每个进程发送的元素数量相同；若各进程片段长度不同，需使用MPI_Gatherv并指定每个进程的发送数量和接收偏移量。
• 内存校验：根进程的接收缓冲区必须足够大，避免内存越界导致程序崩溃。

内容的提问来源于stack exchange，提问作者nicku