teshsuite/smpi/mpich3-test/pt2pt/sendall.c

   1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
   2 /*
   3  *
   4  *  (C) 2007 by Argonne National Laboratory.
   5  *      See COPYRIGHT in top-level directory.
   6  */
   7
   8 #include <stdio.h>
   9 #include <stdlib.h>
  10 #include "mpi.h"
  11 #include "mpitest.h"
  12
  13 /*
  14  * This test makes sure that each process can send to each other process.
  15  * If there are bugs in the handling of request completions or in
  16  * queue operations, then this test may fail on them (it did with
  17  * early EagerShort handling).
  18  */
  19
  20 #define MAXPES 32
  21 #define MYBUFSIZE 16*1024
  22 static int buffer[MAXPES][MYBUFSIZE];
  23
  24 #define NUM_RUNS 10
  25
  26 int main ( int argc, char *argv[] )
  27 {
  28   int i;
  29   int count, size;
  30   int self, npes;
  31   double secs;
  32   MPI_Request request[MAXPES];
  33   MPI_Status status;
  34
  35   MTest_Init (&argc, &argv);
  36   MPI_Comm_rank (MPI_COMM_WORLD, &self);
  37   MPI_Comm_size (MPI_COMM_WORLD, &npes);
  38
  39   if (npes > MAXPES) {
  40     fprintf( stderr, "This program requires a comm_world no larger than %d",
  41              MAXPES );
  42     MPI_Abort( MPI_COMM_WORLD, 1 );
  43     exit(1);
  44   }
  45
  46   for (size = 1; size  <= MYBUFSIZE ; size += size) {
  47       secs = -MPI_Wtime ();
  48       for (count = 0; count < NUM_RUNS; count++) {
  49           MPI_Barrier (MPI_COMM_WORLD);
  50
  51           for (i = 0; i < npes; i++) {
  52               if (i != self)
  53                 MPI_Irecv (buffer[i], size, MPI_INT, i,
  54                          MPI_ANY_TAG, MPI_COMM_WORLD, &request[i]);
  55             }
  56
  57           for (i = 0; i < npes; i++) {
  58               if (i != self)
  59                 MPI_Send (buffer[self], size, MPI_INT, i, 0, MPI_COMM_WORLD);
  60             }
  61
  62           for (i = 0; i < npes; i++) {
  63               if (i != self)
  64                 MPI_Wait (&request[i], &status);
  65             }
  66
  67         }
  68       MPI_Barrier (MPI_COMM_WORLD);
  69       secs += MPI_Wtime ();
  70
  71       if (self == 0) {
  72           secs = secs / (double) NUM_RUNS;
  73           MTestPrintfMsg( 1, "length = %d ints\n", size );
  74         }
  75     }
  76
  77   /* Simple completion is all that we normally ask of this program */
  78
  79   MTest_Finalize( 0 );
  80
  81   MPI_Finalize();
  82   return 0;
  83 }