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     dest = MPI_PROC_NULL;
  37     if (rank == 0) {
  38         dest = size - 1;
  39         a = rank;
  40         b = -1;
  41         MPI_Sendrecv( &a, 1, MPI_INT, dest, 0,
  42                       &b, 1, MPI_INT, dest, 0, newcomm, &status );
  43         if (b != dest) {
  44             errs++;
  45             fprintf( stderr, "Received %d expected %d on %d\n", b, dest, rank );
  46             fflush( stderr );
  47         }
  48         if (status.MPI_SOURCE != dest) {
  49             errs++;
  50             fprintf( stderr, "Source not set correctly in status on %d\n",
  51                      rank );
  52             fflush( stderr );
  53         }
  54     }
  55     else if (rank == size-1) {
  56         dest = 0;
  57         a = rank;
  58         b = -1;
  59         MPI_Sendrecv( &a, 1, MPI_INT, dest, 0,
  60                       &b, 1, MPI_INT, dest, 0, newcomm, &status );
  61         if (b != dest) {
  62             errs++;
  63             fprintf( stderr, "Received %d expected %d on %d\n", b, dest, rank );
  64             fflush( stderr );
  65         }
  66         if (status.MPI_SOURCE != dest) {
  67             errs++;
  68             fprintf( stderr, "Source not set correctly in status on %d\n",
  69                      rank );
  70             fflush( stderr );
  71         }
  72     }
  73
  74     MPI_Comm_free( &newcomm );
  75
  76     MTest_Finalize( errs );
  77
  78     MPI_Finalize();
  79
  80     return 0;
  81 }