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