如何识别MPI进程是否属于给定的Communicator

时间:2012-09-15 15:14:02

标签: mpi

我使用构造MPI_Comm_split()创建了一个通信器,然后我想在 MPI_Comm_World 级别和新通信器级别尝试一些发送/接收通信。创建了两个通信器(一个具有偶数进程排名,另一个具有奇数排名)。

注意:我只需要在包含偶数进程级别的新通信器内尝试通信(虽然进程的级别不是原始进程,但我仍然关注那些进程有他们的记忆空间)

我的代码如下,

    #include<stdio.h>
    #include "mpi.h"
    int main(int arg, char **argv){
    int rank,newrank,newrank2;
    int size, newSize,newSize2;
    int data;
    int recv;
    MPI_Comm newComm, newComm2;
    MPI_Status status, newStatus;
    MPI_Init(&arg,&argv);

    MPI_Comm_size(MPI_COMM_WORLD,&size);
    MPI_Comm_rank(MPI_COMM_WORLD,&rank);    
    //sample data
    data = rank * 100 + 1;

    printf("my rank in the WORLD is: %d\n", rank);
    fflush(stdout); 

    if(rank%2==0){
    MPI_Comm_split(MPI_COMM_WORLD,0,rank,&newComm);
    MPI_Comm_rank(newComm,&newrank);
    MPI_Comm_size(newComm,&newSize);    
    printf("my rank in the newComm1 is : %d\n", newrank);
    }else{      
            MPI_Comm_split(MPI_COMM_WORLD,1,rank,&newComm2);
            MPI_Comm_rank(newComm2,&newrank2);
            MPI_Comm_size(newComm2,&newSize2);        
    printf("my rank in the newComm2 is : %d\n", newrank2);
    }

    //Now try some communication at the newCommunicator and at the WORLD level
    if(rank == 0){
    //sending to process 1 wrt WOLD_Comm
    MPI_Send(&data,1,MPI_INT,1,99,MPI_COMM_WORLD);      
    }
    if(rank==1){
    //Receive from process at WORLD_Comm level. 
    MPI_Recv(&recv,1,MPI_INT,0,99,MPI_COMM_WORLD,&status);
    printf("Received at WORLD Level : %d \n\n",recv);
    }
    if(newSize > 2){
    if(newrank == 2){
        //send to process 1 wrt the newComm
        MPI_Send(&data,1,MPI_INT,1,0,newComm);
    }
    if(newrank==1){
        //Receive from process at newComm level.
        MPI_Recv(&recv,1,MPI_INT,2,0,newComm,&status);
        printf("Received at newComm Level : %d \n\n",recv);
    }
    }
    MPI_Finalize();
    return 0;
    }

当我尝试在new Communicator内发送/接收数据时发生错误。该错误是有道理的,因为不属于新通信器的进程(奇数级进程)不能使用新的通信器句柄(newComm)。

所以我的问题是我如何检查一个给定的进程是否属于新的通信器 所以我可以在做任何沟通之前把这个警卫。

感谢,

1 个答案:

答案 0 :(得分:2)

我将重新审视执行MPI_Comm_split的代码。据我所知,您不需要两个单独的变量newCommnewComm2。当您致电MPI_Comm_split时,您可以想到全局创建多个通信器:每个值color一个。但是每个调用MPI_Comm_split的进程只能获得对一个通信器的引用 - 它所属的通信器。

if MPI_Comm_split语句可以替换为

MPI_Comm_split(MPI_COMM_WORLD, (rank % 2 == 0), rank, &newComm);
MPI_Comm_rank(newComm, &newrank);
MPI_Comm_size(newComm, &newSize);

创建了两个通信器,但newComm指的是不同进程中的不同通信器。

如果奇数编号的进程根本不需要进行通信,则可以为MPI_UNDEFINED提供color。这将使MPI_COMM_NULL的值newComm

MPI_Comm_split(MPI_COMM_WORLD, (rank % 2) ? MPI_UNDEFINED : 0, rank, &newComm);

然后您的代码可以检查(newcomm == MPI_COMM_NULL)