teshsuite/smpi/mpich3-test/coll/coll7.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 #include "mpi.h"
   7 #include <stdio.h>
   8 #include "mpitest.h"
   9
  10 #define MAX_PROCESSES 10
  11
  12 int main( int argc, char **argv )
  13 {
  14     int              rank, size, i,j;
  15     int              table[MAX_PROCESSES][MAX_PROCESSES];
  16     int              errors=0;
  17     int              participants;
  18
  19     MTest_Init( &argc, &argv );
  20     MPI_Comm_rank( MPI_COMM_WORLD, &rank );
  21     MPI_Comm_size( MPI_COMM_WORLD, &size );
  22
  23     /* A maximum of MAX_PROCESSES processes can participate */
  24     if ( size > MAX_PROCESSES ) participants = MAX_PROCESSES;
  25     else              participants = size;
  26     if (MAX_PROCESSES % participants) {
  27         fprintf( stderr, "Number of processors must divide %d\n",
  28                 MAX_PROCESSES );
  29         MPI_Abort( MPI_COMM_WORLD, 1 );
  30         }
  31     /* while (MAX_PROCESSES % participants) participants--; */
  32     if ( (rank < participants) ) {
  33
  34       /* Determine what rows are my responsibility */
  35       int block_size = MAX_PROCESSES / participants;
  36       int begin_row  = rank * block_size;
  37       int end_row    = (rank+1) * block_size;
  38       int send_count = block_size * MAX_PROCESSES;
  39       int recv_count = send_count;
  40
  41       /* Paint my rows my color */
  42       for (i=begin_row; i<end_row ;i++)
  43         for (j=0; j<MAX_PROCESSES; j++)
  44           table[i][j] = rank + 10;
  45
  46       /* Everybody gets the gathered table */
  47       MPI_Allgather(MPI_IN_PLACE, 0, MPI_DATATYPE_NULL,
  48                     &table[0][0], recv_count, MPI_INT, MPI_COMM_WORLD);
  49
  50       /* Everybody should have the same table now,  */
  51       /* This test does not in any way guarantee there are no errors */
  52       /* Print out a table or devise a smart test to make sure it's correct */
  53       for (i=0; i<MAX_PROCESSES;i++) {
  54         if ( (table[i][0] - table[i][MAX_PROCESSES-1] !=0) )
  55           errors++;
  56       }
  57     }
  58
  59     MTest_Finalize( errors );
  60     MPI_Finalize();
  61     return MTestReturnValue( errors );
  62 }