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

   1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
   2 /*
   3  *  (C) 2001 by Argonne National Laboratory.
   4  *      See COPYRIGHT in top-level directory.
   5  */
   6 /*
   7  * Test of reduce scatter.
   8  *
   9  * Each processor contributes its rank + the index to the reduction,
  10  * then receives the ith sum
  11  *
  12  * Can be called with any number of processors.
  13  */
  14
  15 #include "mpi.h"
  16 #include <stdio.h>
  17 #include <stdlib.h>
  18
  19 int main( int argc, char **argv )
  20 {
  21     int      err = 0, toterr;
  22     int      *sendbuf, recvbuf, *recvcounts;
  23     int      size, rank, i, sumval;
  24     MPI_Comm comm;
  25
  26
  27     MPI_Init( &argc, &argv );
  28     comm = MPI_COMM_WORLD;
  29
  30     MPI_Comm_size( comm, &size );
  31     MPI_Comm_rank( comm, &rank );
  32     sendbuf = (int *) malloc( size * sizeof(int) );
  33     for (i=0; i<size; i++)
  34         sendbuf[i] = rank + i;
  35     recvcounts = (int *)malloc( size * sizeof(int) );
  36     for (i=0; i<size; i++)
  37         recvcounts[i] = 1;
  38
  39     MPI_Reduce_scatter( sendbuf, &recvbuf, recvcounts, MPI_INT, MPI_SUM, comm );
  40
  41     sumval = size * rank + ((size - 1) * size)/2;
  42 /* recvbuf should be size * (rank + i) */
  43     if (recvbuf != sumval) {
  44         err++;
  45         fprintf( stdout, "Did not get expected value for reduce scatter\n" );
  46         fprintf( stdout, "[%d] Got %d expected %d\n", rank, recvbuf, sumval );
  47     }
  48
  49     MPI_Allreduce( &err, &toterr, 1, MPI_INT, MPI_SUM, MPI_COMM_WORLD );
  50     if (rank == 0 && toterr == 0) {
  51         printf( " No Errors\n" );
  52     }
  53     MPI_Finalize( );
  54
  55     return toterr;
  56 }