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)
  25         participants = MAX_PROCESSES;
  26     else
  27         participants = size;
  28     if (MAX_PROCESSES % participants) {
  29         fprintf(stderr, "Number of processors must divide %d\n", MAX_PROCESSES);
  30         MPI_Abort(MPI_COMM_WORLD, 1);
  31     }
  32     /* while (MAX_PROCESSES % participants) participants--; */
  33     if ((rank < participants)) {
  34
  35         /* Determine what rows are my responsibility */
  36         int block_size = MAX_PROCESSES / participants;
  37         int begin_row = rank * block_size;
  38         int end_row = (rank + 1) * block_size;
  39         int send_count = block_size * MAX_PROCESSES;
  40         int recv_count = send_count;
  41
  42         /* Paint my rows my color */
  43         for (i = begin_row; i < end_row; i++)
  44             for (j = 0; j < MAX_PROCESSES; j++)
  45                 table[i][j] = rank + 10;
  46
  47         /* Everybody gets the gathered table */
  48         MPI_Allgather(MPI_IN_PLACE, 0, MPI_DATATYPE_NULL,
  49                       &table[0][0], recv_count, MPI_INT, MPI_COMM_WORLD);
  50
  51         /* Everybody should have the same table now,  */
  52         /* This test does not in any way guarantee there are no errors */
  53         /* Print out a table or devise a smart test to make sure it's correct */
  54         for (i = 0; i < MAX_PROCESSES; i++) {
  55             if ((table[i][0] - table[i][MAX_PROCESSES - 1] != 0))
  56                 errors++;
  57         }
  58     }
  59
  60     MTest_Finalize(errors);
  61     MPI_Finalize();
  62     return MTestReturnValue(errors);
  63 }