teshsuite/smpi/mpich3-test/coll/allgather2.c

   1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
   2 /*
   3  *
   4  *  (C) 2003 by Argonne National Laboratory.
   5  *      See COPYRIGHT in top-level directory.
   6  */
   7
   8 #include "mpi.h"
   9 #include "mpitest.h"
  10 #include <stdlib.h>
  11 #include <stdio.h>
  12
  13 /* Gather data from a vector to contiguous.  Use IN_PLACE */
  14
  15 int main(int argc, char **argv)
  16 {
  17     double *vecout;
  18     MPI_Comm comm;
  19     int count, minsize = 2;
  20     int i, errs = 0;
  21     int rank, size;
  22
  23     MTest_Init(&argc, &argv);
  24
  25     while (MTestGetIntracommGeneral(&comm, minsize, 1)) {
  26         if (comm == MPI_COMM_NULL)
  27             continue;
  28         /* Determine the sender and receiver */
  29         MPI_Comm_rank(comm, &rank);
  30         MPI_Comm_size(comm, &size);
  31
  32         for (count = 1; count < 9000; count = count * 2) {
  33             vecout = (double *) malloc(size * count * sizeof(double));
  34
  35             for (i = 0; i < count; i++) {
  36                 vecout[rank * count + i] = rank * count + i;
  37             }
  38             MPI_Allgather(MPI_IN_PLACE, -1, MPI_DATATYPE_NULL, vecout, count, MPI_DOUBLE, comm);
  39             for (i = 0; i < count * size; i++) {
  40                 if (vecout[i] != i) {
  41                     errs++;
  42                     if (errs < 10) {
  43                         fprintf(stderr, "vecout[%d]=%d\n", i, (int) vecout[i]);
  44                     }
  45                 }
  46             }
  47             free(vecout);
  48         }
  49
  50         MTestFreeComm(&comm);
  51     }
  52
  53     MTest_Finalize(errs);
  54     MPI_Finalize();
  55     return 0;
  56 }