teshsuite/smpi/mpich3-test/pt2pt/bsend3.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 <stdio.h>
   7 #include "mpi.h"
   8 #include "mpitest.h"
   9
  10 #define BUFSIZE 2000
  11 int main( int argc, char *argv[] )
  12 {
  13     MPI_Status status;
  14     MPI_Request request;
  15     int a[10], b[10];
  16     int buf[BUFSIZE], *bptr, bl, i, j, rank, size;
  17     int errs = 0;
  18
  19     MTest_Init( 0, 0 );
  20     MPI_Comm_rank( MPI_COMM_WORLD, &rank );
  21     MPI_Comm_size( MPI_COMM_WORLD, &size );
  22     MPI_Buffer_attach( buf, BUFSIZE );
  23
  24     for (j=0; j<10; j++) {
  25         MPI_Bsend_init( a, 10, MPI_INT, 0, 27+j, MPI_COMM_WORLD, &request );
  26         for (i=0; i<10; i++) {
  27             a[i] = (rank + 10 * j) * size + i;
  28         }
  29         MPI_Start( &request );
  30         MPI_Wait( &request, &status );
  31         MPI_Request_free( &request );
  32     }
  33     if (rank == 0) {
  34
  35         for (i=0; i<size; i++) {
  36             for (j=0; j<10; j++) {
  37                 int k;
  38                 status.MPI_TAG = -10;
  39                 status.MPI_SOURCE = -20;
  40                 MPI_Recv( b, 10, MPI_INT, i, 27+j, MPI_COMM_WORLD, &status );
  41
  42                 if (status.MPI_TAG != 27+j) {
  43                     errs++;
  44                     printf( "Wrong tag = %d\n", status.MPI_TAG );
  45                 }
  46                 if (status.MPI_SOURCE != i) {
  47                     errs++;
  48                     printf( "Wrong source = %d\n", status.MPI_SOURCE );
  49                 }
  50                 for (k=0; k<10; k++) {
  51                     if (b[k] != (i + 10 * j) * size + k) {
  52                         errs++;
  53                         printf( "received b[%d] = %d from %d tag %d\n",
  54                                 k, b[k], i, 27+j );
  55                     }
  56                 }
  57             }
  58         }
  59     }
  60     MPI_Buffer_detach( &bptr, &bl );
  61
  62     MTest_Finalize( errs );
  63     MPI_Finalize();
  64     return 0;
  65 }