学习笔记：MPI组通信收集操作MPIBcast，MPIGather与MPIGatherv

最新推荐文章于 2024-10-07 16:44:43 发布

精神科胡主任

最新推荐文章于 2024-10-07 16:44:43 发布

阅读量1.6k

点赞数

分类专栏：学习笔记

本文链接：https://blog.csdn.net/weixin_42742438/article/details/83352082

版权

本文详细介绍了MPI中的组通信操作，包括MPIBcast的广播功能，MPIGather的收集功能，以及MPIGatherv的灵活数据收集特性。MPIGatherv允许从不同进程接收不同数量的数据，并能自定义在接收缓冲区的位置。文中通过实例展示了如何使用MPI_GATHERV进行不同数量数据的收集，并探讨了在实现过程中关于数据数量指定的困惑与解决方案。

摘要由CSDN通过智能技术生成

学习了MPI组通信的概念以及三个组通信函数，在这里写了一些测试代码。广播MPIBcast,收集MPIGather这两个函数不难理解。重点在于更灵活的MPIGatherv函数，使用起来比较麻烦，但显得很有逼格。

MPIGatherv函数可以从不同的进程接收不同数量的数据，所以需要定义一个长度为进程数目的数组，指明从不同进程中收集不同的数据数目。除此之外，还为每一个接收消息在接收缓冲区的位置提供了一个位置偏移displs数组，可以将接收的数据存放到根进程消息缓冲区的任意位置。

MPI_GATHERV明确指出了从不同的进程接收数据元素的个数以及这些数据在 ROOT的接收缓冲区存放的起始位置。

在下面的例子中，我实现了一个整数值的广播，一个整数值和一个数组的收集，还有一个gatherv操作：从每个进程收集进程号+1个数，这些数是进程号本身。

#include<stdio.h>
#include"mpi.h"
#include "stdlib.h"
#define gather_size 10
int main(int argc,char **argv)
{
   
  int rank,proc_size;
  int bcast_value;//广播的值
  int send_gather_num;//收集每个进程的单个值
  int send_gather_list[gather_size];//收集每个进程的多个值
  int *gather_num_recv;//单个值的收集结果
  int *gather_list_recv;//多个值的收集结果
//gatherv
  int gatherv[gather_size];//每个进程还准备gather_size个值
  int *displs;//存放各个进程收集到的数据在收集结果中的存放位置（相对于首地址的偏移）
  int *every_proc_count;//存放从每个进程收集数字的数目。
  int *gatherv_recv;//gatherv的收集结果


  MPI_Init(&argc,&argv);
  printf("before boradcast,bcast_value is %d, in %p\n",bcast_value,&bcast_value);
  MPI_Comm_rank(MPI_COMM_WORLD,&rank);
  MPI_Comm_size(MPI_COMM_WORLD,&proc_size);
  if(rank==0){
   
    scanf("%d",&bcast_value);//输入要广播给各个进程的值。
  }
  MPI_Bcast(&bcast_value,1,MPI_INT,0,MPI_COMM_WORLD);//广播
  printf("after boradcast,bcast_value is %d, in %p\n",bcast_value,&bcast_value);//各进程打印广播后的bcast_value,广播完成。


  send_gather_num=bcast_value*rank;//收集的单个值等于刚才广播的值乘本身进程号
  for(int i=0;i<gather_size;i++)