teshsuite/smpi/mpich3-test/topo/neighb_coll.c

   1 /* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */
   2 /*
   3  *  (C) 2012 by Argonne National Laboratory.
   4  *      See COPYRIGHT in top-level directory.
   5  */
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9
  10 #include <mpi.h>
  11 #include "mpitest.h"
  12
  13 #if !defined(USE_STRICT_MPI) && defined(MPICH)
  14 #define TEST_NEIGHB_COLL 1
  15 #endif
  16
  17 /* assert-like macro that bumps the err count and emits a message */
  18 #define check(x_)                                                                 \
  19     do {                                                                          \
  20         if (!(x_)) {                                                              \
  21             ++errs;                                                               \
  22             if (errs < 10) {                                                      \
  23                 fprintf(stderr, "check failed: (%s), line %d\n", #x_, __LINE__); \
  24             }                                                                     \
  25         }                                                                         \
  26     } while (0)
  27
  28 int main(int argc, char *argv[])
  29 {
  30     int errs = 0;
  31     int wrank, wsize;
  32     int periods[1] = { 0 };
  33     MPI_Comm cart, dgraph, graph;
  34
  35     MPI_Init(&argc, &argv);
  36     MPI_Comm_rank(MPI_COMM_WORLD, &wrank);
  37     MPI_Comm_size(MPI_COMM_WORLD, &wsize);
  38
  39 #if defined(TEST_NEIGHB_COLL)
  40     /* a basic test for the 10 (5 patterns x {blocking,nonblocking}) MPI-3
  41      * neighborhood collective routines */
  42
  43     /* (wrap)--> 0 <--> 1 <--> ... <--> p-1 <--(wrap) */
  44     MPI_Cart_create(MPI_COMM_WORLD, 1, &wsize, periods, /*reorder= */ 0, &cart);
  45
  46     /* allgather */
  47     {
  48         int sendbuf[1] = { wrank };
  49         int recvbuf[2] = { 0xdeadbeef, 0xdeadbeef };
  50
  51         /* should see one send to each neighbor (rank-1 and rank+1) and one receive
  52          * each from same */
  53         MPI_Neighbor_allgather(sendbuf, 1, MPI_INT, recvbuf, 1, MPI_INT, cart);
  54
  55         if (wrank == 0)
  56             check(recvbuf[0] == 0xdeadbeef);
  57         else
  58             check(recvbuf[0] == wrank - 1);
  59
  60         if (wrank == wsize - 1)
  61             check(recvbuf[1] == 0xdeadbeef);
  62         else
  63             check(recvbuf[1] == wrank + 1);
  64     }
  65
  66     /* allgatherv */
  67     {
  68         int sendbuf[1] = { wrank };
  69         int recvbuf[2] = { 0xdeadbeef, 0xdeadbeef };
  70         int recvcounts[2] = { 1, 1 };
  71         int displs[2] = { 1, 0 };
  72
  73         /* should see one send to each neighbor (rank-1 and rank+1) and one receive
  74          * each from same, but put them in opposite slots in the buffer */
  75         MPI_Neighbor_allgatherv(sendbuf, 1, MPI_INT, recvbuf, recvcounts, displs, MPI_INT, cart);
  76
  77         if (wrank == 0)
  78             check(recvbuf[1] == 0xdeadbeef);
  79         else
  80             check(recvbuf[1] == wrank - 1);
  81
  82         if (wrank == wsize - 1)
  83             check(recvbuf[0] == 0xdeadbeef);
  84         else
  85             check(recvbuf[0] == wrank + 1);
  86     }
  87
  88     /* alltoall */
  89     {
  90         int sendbuf[2] = { -(wrank + 1), wrank + 1 };
  91         int recvbuf[2] = { 0xdeadbeef, 0xdeadbeef };
  92
  93         /* should see one send to each neighbor (rank-1 and rank+1) and one
  94          * receive each from same */
  95         MPI_Neighbor_alltoall(sendbuf, 1, MPI_INT, recvbuf, 1, MPI_INT, cart);
  96
  97         if (wrank == 0)
  98             check(recvbuf[0] == 0xdeadbeef);
  99         else
 100             check(recvbuf[0] == wrank);
 101
 102         if (wrank == wsize - 1)
 103             check(recvbuf[1] == 0xdeadbeef);
 104         else
 105             check(recvbuf[1] == -(wrank + 2));
 106     }
 107
 108     /* alltoallv */
 109     {
 110         int sendbuf[2] = { -(wrank + 1), wrank + 1 };
 111         int recvbuf[2] = { 0xdeadbeef, 0xdeadbeef };
 112         int sendcounts[2] = { 1, 1 };
 113         int recvcounts[2] = { 1, 1 };
 114         int sdispls[2] = { 0, 1 };
 115         int rdispls[2] = { 1, 0 };
 116
 117         /* should see one send to each neighbor (rank-1 and rank+1) and one receive
 118          * each from same, but put them in opposite slots in the buffer */
 119         MPI_Neighbor_alltoallv(sendbuf, sendcounts, sdispls, MPI_INT,
 120                                recvbuf, recvcounts, rdispls, MPI_INT, cart);
 121
 122         if (wrank == 0)
 123             check(recvbuf[1] == 0xdeadbeef);
 124         else
 125             check(recvbuf[1] == wrank);
 126
 127         if (wrank == wsize - 1)
 128             check(recvbuf[0] == 0xdeadbeef);
 129         else
 130             check(recvbuf[0] == -(wrank + 2));
 131     }
 132
 133     /* alltoallw */
 134     {
 135         int sendbuf[2] = { -(wrank + 1), wrank + 1 };
 136         int recvbuf[2] = { 0xdeadbeef, 0xdeadbeef };
 137         int sendcounts[2] = { 1, 1 };
 138         int recvcounts[2] = { 1, 1 };
 139         MPI_Aint sdispls[2] = { 0, sizeof(int) };
 140         MPI_Aint rdispls[2] = { sizeof(int), 0 };
 141         MPI_Datatype sendtypes[2] = { MPI_INT, MPI_INT };
 142         MPI_Datatype recvtypes[2] = { MPI_INT, MPI_INT };
 143
 144         /* should see one send to each neighbor (rank-1 and rank+1) and one receive
 145          * each from same, but put them in opposite slots in the buffer */
 146         MPI_Neighbor_alltoallw(sendbuf, sendcounts, sdispls, sendtypes,
 147                                recvbuf, recvcounts, rdispls, recvtypes, cart);
 148
 149         if (wrank == 0)
 150             check(recvbuf[1] == 0xdeadbeef);
 151         else
 152             check(recvbuf[1] == wrank);
 153
 154         if (wrank == wsize - 1)
 155             check(recvbuf[0] == 0xdeadbeef);
 156         else
 157             check(recvbuf[0] == -(wrank + 2));
 158     }
 159
 160
 161     MPI_Comm_free(&cart);
 162 #endif /* defined(TEST_NEIGHB_COLL) */
 163
 164     MPI_Reduce((wrank == 0 ? MPI_IN_PLACE : &errs), &errs, 1, MPI_INT, MPI_SUM, 0, MPI_COMM_WORLD);
 165     if (wrank == 0) {
 166         if (errs) {
 167             printf("found %d errors\n", errs);
 168         }
 169         else {
 170             printf(" No errors\n");
 171         }
 172     }
 173     MPI_Finalize();
 174
 175     return 0;
 176 }