MPI:多个重叠的传播者

时间:2016-12-31 02:11:12

标签: mpi communicator

我想创建MPI通信器,将进程与等级0链接到其他每个进程。假设n是进程的总数。然后,具有等级0的过程应该具有n-1个通信器,而每个其他过程具有一个通信器。这是否可能,如果是,为什么我不能使用下面的程序来实现这个目标呢?

使用mpic ++编译器编译下面的代码终止,而我的计算机上没有警告和错误。但是当我使用3个或更多进程(mpiexec -n 3)运行生成的程序时,它永远不会终止。

可能我误解了MPI中传播者的概念。也许有人可以帮助我理解为什么下面的程序会被卡住,以及创建这些传播者的更好方法是什么?感谢。

#include <iostream>
#include <vector>
#include <thread>
#include <chrono>
#include "mpi.h"
void FinalizeMPI();
void InitMPI(int argc, char** argv);
int main(int argc, char** argv) {
    InitMPI(argc, argv);

    int rank,comm_size;
    MPI_Comm_rank(MPI_COMM_WORLD,&rank);
    MPI_Comm_size(MPI_COMM_WORLD,&comm_size);

    if(comm_size<2) {
        FinalizeMPI();
        return 0;
    }

    MPI_Group GroupAll;
    MPI_Comm_group(MPI_COMM_WORLD, &GroupAll);

    if(rank==0) {
        std::vector<MPI_Group> myGroups(comm_size-1);
        std::vector<MPI_Comm> myComms(comm_size-1);
        for(int k=1;k<comm_size;++k) {
           int ranks[2]{0, k};
           MPI_Group_incl(GroupAll, 2, ranks, &myGroups[k-1]);
           int err = MPI_Comm_create(MPI_COMM_WORLD, myGroups[k-1], &myComms[k-1]);
           std::cout << "Error: " << err << std::endl;
        }
    } else {
        MPI_Group myGroup;
        MPI_Comm myComm;
        int ranks[2]{0,rank};
        MPI_Group_incl(GroupAll, 2, ranks, &myGroup);
        int err = MPI_Comm_create(MPI_COMM_WORLD, myGroup, &myComm);
        std::cout << "Error: " << err << std::endl;
    }
    std::cout << "Communicators created: " << rank << std::endl;
    std::this_thread::sleep_for(std::chrono::milliseconds(1000));

    FinalizeMPI();
    return 0;
}

void FinalizeMPI() {
   int flag;
   MPI_Finalized(&flag);
   if(!flag)
      MPI_Finalize();
}

void InitMPI(int argc, char** argv) {
   int flag;
   MPI_Initialized(&flag);
   if(!flag) {
      int provided_Support;
      MPI_Init_thread(&argc, &argv, MPI_THREAD_MULTIPLE, &provided_Support);
      if(provided_Support!=MPI_THREAD_MULTIPLE) {
          exit(0);
      }
   }
}

1 个答案:

答案 0 :(得分:1)

MPI_Comm_create是初始通信器(MPI_COMM_WORLD)上的集合操作 - 您必须在所有进程上调用它。

解决问题的最简单方法是按照您的方式使用MPI_Comm_create_group。它类似于MPI_Comm_create,除了它是集体的集体。