Blame test/mpi/comm/comm_create_group.c

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
}