Blame test/mpi/comm/comm_group_rand.c

Packit 0848f5
/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
Packit 0848f5
/*
Packit 0848f5
 *  (C) 2003 by Argonne National Laboratory.
Packit 0848f5
 *      See COPYRIGHT in top-level directory.
Packit 0848f5
 */
Packit 0848f5
Packit 0848f5
#include <stdio.h>
Packit 0848f5
#include <stdlib.h>
Packit 0848f5
#include <mpi.h>
Packit 0848f5
#include "mpitestconf.h"
Packit 0848f5
Packit 0848f5
#define LOOPS 100
Packit 0848f5
Packit 0848f5
int main(int argc, char **argv)
Packit 0848f5
{
Packit 0848f5
    int rank, size, i, j, count;
Packit 0848f5
    MPI_Group full_group, sub_group;
Packit 0848f5
    int *included, *ranks;
Packit 0848f5
    MPI_Comm comm;
Packit 0848f5
Packit 0848f5
    MPI_Init(NULL, NULL);
Packit 0848f5
    MPI_Comm_rank(MPI_COMM_WORLD, &rank;;
Packit 0848f5
    MPI_Comm_size(MPI_COMM_WORLD, &size);
Packit 0848f5
Packit 0848f5
    ranks = malloc(size * sizeof(int));
Packit 0848f5
    included = malloc(size * sizeof(int));
Packit 0848f5
    MPI_Comm_group(MPI_COMM_WORLD, &full_group);
Packit 0848f5
Packit 0848f5
    for (j = 0; j < LOOPS; j++) {
Packit 0848f5
        srand(j);       /* Deterministic seed */
Packit 0848f5
Packit 0848f5
        count = 0;
Packit 0848f5
        for (i = 0; i < size; i++) {
Packit 0848f5
            if (rand() % 2) {   /* randomly include a rank */
Packit 0848f5
                included[i] = 1;
Packit 0848f5
                ranks[count++] = i;
Packit 0848f5
            }
Packit 0848f5
            else
Packit 0848f5
                included[i] = 0;
Packit 0848f5
        }
Packit 0848f5
Packit 0848f5
        MPI_Group_incl(full_group, count, ranks, &sub_group);
Packit 0848f5
Packit 0848f5
        if (included[rank]) {
Packit 0848f5
            MPI_Comm_create_group(MPI_COMM_WORLD, sub_group, 0, &comm);
Packit 0848f5
            MPI_Barrier(comm);
Packit 0848f5
            MPI_Comm_free(&comm);
Packit 0848f5
        }
Packit 0848f5
Packit 0848f5
        MPI_Group_free(&sub_group);
Packit 0848f5
    }
Packit 0848f5
Packit 0848f5
    MPI_Group_free(&full_group);
Packit 0848f5
Packit 0848f5
    if (rank == 0)
Packit 0848f5
        printf(" No Errors\n");
Packit 0848f5
Packit 0848f5
    free(ranks);
Packit 0848f5
    free(included);
Packit 0848f5
    MPI_Finalize();
Packit 0848f5
Packit 0848f5
    return 0;
Packit 0848f5
}