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   }
  44
  45   for (size = 1; size  <= MYBUFSIZE ; size += size) {
  46       secs = -MPI_Wtime ();
  47       for (count = 0; count < NUM_RUNS; count++) {
  48           MPI_Barrier (MPI_COMM_WORLD);
  49
  50           for (i = 0; i < npes; i++) {
  51               if (i != self)
  52                 MPI_Irecv (buffer[i], size, MPI_INT, i,
  53                          MPI_ANY_TAG, MPI_COMM_WORLD, &request[i]);
  54             }
  55
  56           for (i = 0; i < npes; i++) {
  57               if (i != self)
  58                 MPI_Send (buffer[self], size, MPI_INT, i, 0, MPI_COMM_WORLD);
  59             }
  60
  61           for (i = 0; i < npes; i++) {
  62               if (i != self)
  63                 MPI_Wait (&request[i], &status);
  64             }
  65
  66         }
  67       MPI_Barrier (MPI_COMM_WORLD);
  68       secs += MPI_Wtime ();
  69
  70       if (self == 0) {
  71           secs = secs / (double) NUM_RUNS;
  72           MTestPrintfMsg( 1, "length = %d ints\n", size );
  73         }
  74     }
  75
  76   /* Simple completion is all that we normally ask of this program */
  77
  78   MTest_Finalize( 0 );
  79
  80   MPI_Finalize();
  81   return 0;
  82 }