teshsuite/smpi/mpich3-test/comm/dup.c

   1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
   2 /*
   3  *
   4  *  (C) 2003 by Argonne National Laboratory.
   5  *      See COPYRIGHT in top-level directory.
   6  */
   7 #include "mpi.h"
   8 #include <stdio.h>
   9 #include "mpitest.h"
  10
  11 int main( int argc, char **argv )
  12 {
  13     int errs = 0;
  14     int rank, size, wrank, wsize, dest, a, b;
  15     MPI_Comm newcomm;
  16     MPI_Status status;
  17
  18     MTest_Init( &argc, &argv );
  19
  20     /* Can we run comm dup at all? */
  21     MPI_Comm_dup( MPI_COMM_WORLD, &newcomm );
  22
  23     /* Check basic properties */
  24     MPI_Comm_size( MPI_COMM_WORLD, &wsize );
  25     MPI_Comm_rank( MPI_COMM_WORLD, &wrank );
  26     MPI_Comm_size( newcomm, &size );
  27     MPI_Comm_rank( newcomm, &rank );
  28
  29     if (size != wsize || rank != wrank) {
  30         errs++;
  31         fprintf( stderr, "Size (%d) or rank (%d) wrong\n", size, rank );
  32         fflush( stderr );
  33     }
  34
  35     /* Can we communicate with this new communicator? */
  36     if (rank == 0) {
  37         dest = size - 1;
  38         a = rank;
  39         b = -1;
  40         MPI_Sendrecv( &a, 1, MPI_INT, dest, 0,
  41                       &b, 1, MPI_INT, dest, 0, newcomm, &status );
  42         if (b != dest) {
  43             errs++;
  44             fprintf( stderr, "Received %d expected %d on %d\n", b, dest, rank );
  45             fflush( stderr );
  46         }
  47         if (status.MPI_SOURCE != dest) {
  48             errs++;
  49             fprintf( stderr, "Source not set correctly in status on %d\n",
  50                      rank );
  51             fflush( stderr );
  52         }
  53     }
  54     else if (rank == size-1) {
  55         dest = 0;
  56         a = rank;
  57         b = -1;
  58         MPI_Sendrecv( &a, 1, MPI_INT, dest, 0,
  59                       &b, 1, MPI_INT, dest, 0, newcomm, &status );
  60         if (b != dest) {
  61             errs++;
  62             fprintf( stderr, "Received %d expected %d on %d\n", b, dest, rank );
  63             fflush( stderr );
  64         }
  65         if (status.MPI_SOURCE != dest) {
  66             errs++;
  67             fprintf( stderr, "Source not set correctly in status on %d\n",
  68                      rank );
  69             fflush( stderr );
  70         }
  71     }
  72
  73     MPI_Comm_free( &newcomm );
  74
  75     MTest_Finalize( errs );
  76
  77     MPI_Finalize();
  78
  79     return 0;
  80 }