Вы можете переопределить реализацию MPI_Comm_split
и MPI_Comm_free
для ручного подсчета создания и уничтожения коммуникатора.
Вот простой пример
Переопределение MPI_Comm_split
и MPI_Comm_free
#include "mpi.h"
#include "stdio.h"
static int comm_counter=0;
int MPI_Comm_split(MPI_Comm comm, int color, int key, MPI_Comm *newcomm)
{
int world_rank;
MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
comm_counter++;
printf("%s %i %s %i\n", "MPI_Comm_split ", comm_counter, " from ", world_rank);
return PMPI_Comm_split(comm, color, key, newcomm);
}
int MPI_Comm_free(MPI_Comm *comm)
{
int world_rank;
MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
comm_counter--;
printf("%s %i %s %i\n", "PMPI_Comm_free ", comm_counter, " from ", world_rank);
return PMPI_Comm_free(comm);
}
Скомпилируйте этот код для связи.
В моем случае я сделал mpicc -c comm_split.c -o comm_split.o
Ваш код остался нетронутым. Вы можете использовать его без других модификаций.
Простой пример основной программы с использованием MPI_Comm_split
и MPI_Comm_free
C ++ case
#include "mpi.h"
int main()
{
MPI_Init(NULL, NULL);
// Get the rank and size in the original communicator
int world_rank, world_size;
MPI_Comm_rank(MPI_COMM_WORLD, &world_rank);
MPI_Comm_size(MPI_COMM_WORLD, &world_size);
int color = world_rank / 4; // Determine color based on row
// Split the communicator based on the color and use the
// original rank for ordering
MPI_Comm row_comm, row_comm2;
MPI_Comm_split(MPI_COMM_WORLD, color, world_rank, &row_comm);
MPI_Comm_split(MPI_COMM_WORLD, color, world_rank, &row_comm2);
int row_rank, row_size;
MPI_Comm_rank(row_comm, &row_rank);
MPI_Comm_size(row_comm, &row_size);
printf("WORLD RANK/SIZE: %d/%d \t ROW RANK/SIZE: %d/%d\n",
world_rank, world_size, row_rank, row_size);
MPI_Comm_free(&row_comm);
MPI_Finalize();
}
Корпус Fortran
program test
include "mpif.h"
integer comm_world, group_world, new_comm, new_comm2, ierr
integer world_rank, world_size;
integer color
call MPI_INIT(ierr)
comm_world = MPI_COMM_WORLD
call MPI_Comm_rank(comm_world, world_rank, ierr);
color = world_rank / 4
call MPI_Comm_split(comm_world, color, world_rank, new_comm, ierr)
call MPI_Comm_split(comm_world, color, world_rank,
& new_comm2, ierr)
call MPI_Comm_free(new_comm, ierr)
call MPI_Finalize(ierr)
end program
Компиляция + ссылка с переопределением MPI_Comm_split
и MPI_Comm_free
mpif77 test.f comm_split.o
mpiCC test.cpp comm_split.o
Для случая Fortran вы получите что-то вроде
MPI_Comm_split 1 from 3
MPI_Comm_split 1 from 0
MPI_Comm_split 1 from 1
MPI_Comm_split 1 from 2
MPI_Comm_split 2 from 0
PMPI_Comm_free 1 from 0
MPI_Comm_split 2 from 1
PMPI_Comm_free 1 from 1
MPI_Comm_split 2 from 2
PMPI_Comm_free 1 from 2
MPI_Comm_split 2 from 3
PMPI_Comm_free 1 from 3
Который дает вам информацию о количестве коммуникаторов, вовлеченных в каждый процесс.