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

学习了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++)
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值