src/smpi/colls/reduce-flat-tree.c

   1 #include "colls_private.h"
   2 //#include <star-reduction.c>
   3
   4 int
   5 smpi_coll_tuned_reduce_flat_tree(void *sbuf, void *rbuf, int count,
   6                                  MPI_Datatype dtype, MPI_Op op,
   7                                  int root, MPI_Comm comm)
   8 {
   9   int i, tag = COLL_TAG_REDUCE;
  10   int size;
  11   int rank;
  12   MPI_Aint extent;
  13   char *origin = 0;
  14   char *inbuf;
  15   MPI_Status status;
  16
  17   rank = smpi_comm_rank(comm);
  18   size = smpi_comm_size(comm);
  19
  20   /* If not root, send data to the root. */
  21   extent = smpi_datatype_get_extent(dtype);
  22
  23   if (rank != root) {
  24     smpi_mpi_send(sbuf, count, dtype, root, tag, comm);
  25     return 0;
  26   }
  27
  28   /* Root receives and reduces messages.  Allocate buffer to receive
  29      messages. */
  30
  31   if (size > 1)
  32     origin = (char *) xbt_malloc(count * extent);
  33
  34
  35   /* Initialize the receive buffer. */
  36   if (rank == (size - 1))
  37     smpi_mpi_sendrecv(sbuf, count, dtype, rank, tag,
  38                  rbuf, count, dtype, rank, tag, comm, &status);
  39   else
  40     smpi_mpi_recv(rbuf, count, dtype, size - 1, tag, comm, &status);
  41
  42   /* Loop receiving and calling reduction function (C or Fortran). */
  43
  44   for (i = size - 2; i >= 0; --i) {
  45     if (rank == i)
  46       inbuf = sbuf;
  47     else {
  48       smpi_mpi_recv(origin, count, dtype, i, tag, comm, &status);
  49       inbuf = origin;
  50     }
  51
  52     /* Call reduction function. */
  53     smpi_op_apply(op, inbuf, rbuf, &count, &dtype);
  54
  55   }
  56
  57   if (origin)
  58     free(origin);
  59
  60   /* All done */
  61   return 0;
  62 }