teshsuite/smpi/mpich3-test/coll/allgatherv2.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.  This is
  14    the trivial version based on the allgather test (allgatherv but with
  15    constant data sizes) */
  16
  17 int main(int argc, char **argv)
  18 {
  19     double *vecout;
  20     MPI_Comm comm;
  21     int count, minsize = 2;
  22     int i, errs = 0;
  23     int rank, size;
  24     int *displs, *recvcounts;
  25
  26     MTest_Init(&argc, &argv);
  27
  28     while (MTestGetIntracommGeneral(&comm, minsize, 1)) {
  29         if (comm == MPI_COMM_NULL)
  30             continue;
  31         /* Determine the sender and receiver */
  32         MPI_Comm_rank(comm, &rank);
  33         MPI_Comm_size(comm, &size);
  34
  35         displs = (int *) malloc(size * sizeof(int));
  36         recvcounts = (int *) malloc(size * sizeof(int));
  37
  38         for (count = 1; count < 9000; count = count * 2) {
  39             vecout = (double *) malloc(size * count * sizeof(double));
  40
  41             for (i = 0; i < count; i++) {
  42                 vecout[rank * count + i] = rank * count + i;
  43             }
  44             for (i = 0; i < size; i++) {
  45                 recvcounts[i] = count;
  46                 displs[i] = i * count;
  47             }
  48             MPI_Allgatherv(MPI_IN_PLACE, -1, MPI_DATATYPE_NULL,
  49                            vecout, recvcounts, displs, MPI_DOUBLE, comm);
  50             for (i = 0; i < count * size; i++) {
  51                 if (vecout[i] != i) {
  52                     errs++;
  53                     if (errs < 10) {
  54                         fprintf(stderr, "vecout[%d]=%d\n", i, (int) vecout[i]);
  55                     }
  56                 }
  57             }
  58             free(vecout);
  59         }
  60
  61         free(displs);
  62         free(recvcounts);
  63         MTestFreeComm(&comm);
  64     }
  65
  66     MTest_Finalize(errs);
  67     MPI_Finalize();
  68     return 0;
  69 }