如何知道作为MPI中通信器一部分的处理器的所有级别?

时间:2013-02-04 15:58:24

标签: mpi

无论如何,我可以通过它了解作为沟通者一部分的所有过程吗? 假设总共有16个MPI进程,而MPI_Comm comm有4个进程作为一个组。鉴于沟通者沟通,我们能否知道作为沟通者一部分的所有流程?

谢谢

1 个答案:

答案 0 :(得分:4)

每个通信器都有一个关联的进程组,可以通过调用{绑定中的MPI_COMM_GROUPMPI_Comm_group)来获得。获得comm的流程组后,可以使用MPI_GROUP_TRANSLATE_RANKScomm组中的排名列表转换为MPI_COMM_WORLD组中的相应排名。必须经历翻译流程,因为在comm组中,参与流程的排名范围从0MPI_COMM_SIZE(comm)-1

以下是一个示例实现:

void print_comm_ranks(MPI_Comm comm)
{
   MPI_Group grp, world_grp;

   MPI_Comm_group(MPI_COMM_WORLD, &world_grp);
   MPI_Comm_group(comm, &grp);

   int grp_size;

   MPI_Group_size(grp, &grp_size);

   int *ranks = malloc(grp_size * sizeof(int));
   int *world_ranks = malloc(grp_size * sizeof(int));

   for (int i = 0; i < grp_size; i++)
      ranks[i] = i;

   MPI_Group_translate_ranks(grp, grp_size, ranks, world_grp, world_ranks);

   for (int i = 0; i < grp_size; i++)
      printf("comm[%d] has world rank %d\n", i, world_ranks[i]);

   free(ranks); free(world_ranks);

   MPI_Group_free(&grp);
   MPI_Group_free(&world_grp);
}

以下是一个示例用法:

int rank;
MPI_Comm comm;

MPI_Comm_rank(MPI_COMM_WORLD, &rank);
MPI_Comm_split(MPI_COMM_WORLD, rank % 2, rank, &comm);

if (rank == 0)
{
   printf("Rank 0 view:\n");
   print_comm_ranks(comm);
}
else if (rank == 1)
{
   printf("Rank 1 view:\n");
   print_comm_ranks(comm);
}

和7个进程的相应输出:

Rank 0 view:
comm[0] has world rank 0
comm[1] has world rank 2
comm[2] has world rank 4
comm[3] has world rank 6
Rank 1 view:
comm[0] has world rank 1
comm[1] has world rank 3
comm[2] has world rank 5

(排名01在分割后会在不同的传播者中结束)

请注意,您只能枚举当前进程知道的通信器的内容,因为通信器由其句柄引用,并且这些是每个进程的本地值。

相关问题