将同一节点上运行的进程划分到同一个通信域

在MPI中,需要将同一个节点上运行的程序划分到同一个通行域中,这样可以使得消息的传递更加快捷和方便,实现代码如下:

#include "mpi.h"

#include <stdio.h>
#include <unistd.h>
#include <string.h>

int main(int argc,char ** argv)
{
   int rank;
   int size;
   char myName[100];
   int i=0;
   
   
   MPI_Init(&argc,&argv);
   MPI_Comm_rank(MPI_COMM_WORLD,&rank);
   MPI_Comm_size(MPI_COMM_WORLD,&size);
// printf("%d\n",size);
   char *allHostName=(char *)malloc(size*sizeof(MPI_CHAR)*100);
   gethostname(myName,100);
   
 
   MPI_Allgather(myName,100,MPI_CHAR,allHostName,100,MPI_CHAR,MPI_COMM_WORLD);
 
   int color,key;
   int count=0;
   MPI_Comm myWorld,splitWorld;
   for(i=0;i<size;i++)
    {
         if(strcmp(myName,allHostName+100*i)==0)
           {
              if(count==0)
                  key=i;
              if(i==rank)
                 {
                   color=count;
                   break;
                 }
               count++;
             
           }
    }
    MPI_Comm_dup(MPI_COMM_WORLD,&myWorld);
    MPI_Comm_split(myWorld,color,key,&splitWorld);
    
    MPI_Finalize();
 
   return 0; 

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值