|
Packit Service |
c5cf8c |
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
|
|
Packit Service |
c5cf8c |
/*
|
|
Packit Service |
c5cf8c |
* (C) 2011 by Argonne National Laboratory.
|
|
Packit Service |
c5cf8c |
* See COPYRIGHT in top-level directory.
|
|
Packit Service |
c5cf8c |
*/
|
|
Packit Service |
c5cf8c |
#include "mpi.h"
|
|
Packit Service |
c5cf8c |
#include <stdio.h>
|
|
Packit Service |
c5cf8c |
#include <string.h>
|
|
Packit Service |
c5cf8c |
#include <stdlib.h>
|
|
Packit Service |
c5cf8c |
#include <assert.h>
|
|
Packit Service |
c5cf8c |
#include "mpitest.h"
|
|
Packit Service |
c5cf8c |
|
|
Packit Service |
c5cf8c |
int main(int argc, char *argv[])
|
|
Packit Service |
c5cf8c |
{
|
|
Packit Service |
c5cf8c |
int size, rank, i, *excl;
|
|
Packit Service |
c5cf8c |
MPI_Group world_group, even_group;
|
|
Packit Service |
c5cf8c |
MPI_Comm even_comm;
|
|
Packit Service |
c5cf8c |
|
|
Packit Service |
c5cf8c |
MTest_Init(&argc, &argv);
|
|
Packit Service |
c5cf8c |
|
|
Packit Service |
c5cf8c |
MPI_Comm_size(MPI_COMM_WORLD, &size);
|
|
Packit Service |
c5cf8c |
MPI_Comm_rank(MPI_COMM_WORLD, &rank;;
|
|
Packit Service |
c5cf8c |
|
|
Packit Service |
c5cf8c |
if (size % 2) {
|
|
Packit Service |
c5cf8c |
fprintf(stderr, "this program requires a multiple of 2 number of processes\n");
|
|
Packit Service |
c5cf8c |
MPI_Abort(MPI_COMM_WORLD, 1);
|
|
Packit Service |
c5cf8c |
}
|
|
Packit Service |
c5cf8c |
|
|
Packit Service |
c5cf8c |
excl = malloc((size / 2) * sizeof(int));
|
|
Packit Service |
c5cf8c |
assert(excl);
|
|
Packit Service |
c5cf8c |
|
|
Packit Service |
c5cf8c |
/* exclude the odd ranks */
|
|
Packit Service |
c5cf8c |
for (i = 0; i < size / 2; i++)
|
|
Packit Service |
c5cf8c |
excl[i] = (2 * i) + 1;
|
|
Packit Service |
c5cf8c |
|
|
Packit Service |
c5cf8c |
/* Create some groups */
|
|
Packit Service |
c5cf8c |
MPI_Comm_group(MPI_COMM_WORLD, &world_group);
|
|
Packit Service |
c5cf8c |
MPI_Group_excl(world_group, size / 2, excl, &even_group);
|
|
Packit Service |
c5cf8c |
MPI_Group_free(&world_group);
|
|
Packit Service |
c5cf8c |
|
|
Packit Service |
c5cf8c |
if (rank % 2 == 0) {
|
|
Packit Service |
c5cf8c |
/* Even processes create a group for themselves */
|
|
Packit Service |
c5cf8c |
MPI_Comm_create_group(MPI_COMM_WORLD, even_group, 0, &even_comm);
|
|
Packit Service |
c5cf8c |
MPI_Barrier(even_comm);
|
|
Packit Service |
c5cf8c |
MPI_Comm_free(&even_comm);
|
|
Packit Service |
c5cf8c |
}
|
|
Packit Service |
c5cf8c |
|
|
Packit Service |
c5cf8c |
MPI_Group_free(&even_group);
|
|
Packit Service |
c5cf8c |
MPI_Barrier(MPI_COMM_WORLD);
|
|
Packit Service |
c5cf8c |
|
|
Packit Service |
c5cf8c |
free(excl);
|
|
Packit Service |
c5cf8c |
MTest_Finalize(0);
|
|
Packit Service |
c5cf8c |
return 0;
|
|
Packit Service |
c5cf8c |
}
|