src/smpi/colls/colls.h

   1 #ifndef SMPI_COLLS_H
   2 #define SMPI_COLLS_H
   3
   4 #include <math.h>
   5 #include "smpi/mpi.h"
   6 #include "smpi/private.h"
   7 #include "xbt.h"
   8
   9 void star_reduction(MPI_Op op, void *src, void *target, int *count, MPI_Datatype *dtype);
  10
  11 #define COLL_DESCRIPTION(cat, ret, args, name) \
  12   {# name,\
  13    # cat " " # name " collective",\
  14    smpi_coll_tuned_ ## cat ## _ ## name}
  15
  16 #define COLL_PROTO(cat, ret, args, name) \
  17   ret smpi_coll_tuned_ ## cat ## _ ## name(COLL_UNPAREN args);
  18 #define COLL_UNPAREN(...)  __VA_ARGS__
  19
  20 #define COLL_APPLY(action, sig, name) action(sig, name)
  21 #define COLL_COMMA ,
  22 #define COLL_NOsep
  23 #define COLL_NOTHING(...)
  24
  25
  26 /*************
  27  * ALLGATHER *
  28  *************/
  29 #define COLL_ALLGATHER_SIG allgather, int, \
  30                           (void *send_buff, int send_count, MPI_Datatype send_type, \
  31                            void *recv_buff, int recv_count, MPI_Datatype recv_type, \
  32                            MPI_Comm comm)
  33
  34 #define COLL_ALLGATHERS(action, COLL_sep) \
  35 COLL_NOTHING(COLL_APPLY(action, COLL_ALLGATHER_SIG, 2dmesh) COLL_sep) \
  36 COLL_NOTHING(COLL_APPLY(action, COLL_ALLGATHER_SIG, 3dmesh) COLL_sep) \
  37 COLL_NOTHING(COLL_APPLY(action, COLL_ALLGATHER_SIG, bruck) COLL_sep) \
  38 COLL_APPLY(action, COLL_ALLGATHER_SIG, GB) COLL_sep \
  39 COLL_APPLY(action, COLL_ALLGATHER_SIG, loosely_lr) COLL_sep \
  40 COLL_APPLY(action, COLL_ALLGATHER_SIG, lr) COLL_sep \
  41 COLL_APPLY(action, COLL_ALLGATHER_SIG, NTSLR) COLL_sep \
  42 COLL_APPLY(action, COLL_ALLGATHER_SIG, NTSLR_NB) COLL_sep \
  43 COLL_APPLY(action, COLL_ALLGATHER_SIG, pair) COLL_sep \
  44 COLL_APPLY(action, COLL_ALLGATHER_SIG, rdb) COLL_sep \
  45 COLL_APPLY(action, COLL_ALLGATHER_SIG, RDB) COLL_sep \
  46 COLL_APPLY(action, COLL_ALLGATHER_SIG, rhv) COLL_sep \
  47 COLL_APPLY(action, COLL_ALLGATHER_SIG, ring) COLL_sep \
  48 COLL_APPLY(action, COLL_ALLGATHER_SIG, SMP_NTS) COLL_sep \
  49 COLL_APPLY(action, COLL_ALLGATHER_SIG, smp_simple) COLL_sep \
  50 COLL_APPLY(action, COLL_ALLGATHER_SIG, SMP_simple) COLL_sep \
  51 COLL_APPLY(action, COLL_ALLGATHER_SIG, spreading_simple)
  52
  53 COLL_ALLGATHERS(COLL_PROTO, COLL_NOsep)
  54
  55
  56 /*************
  57  * ALLREDUCE *
  58  *************/
  59 #define COLL_ALLREDUCE_SIG allreduce, int, \
  60                           (void *sbuf, void *rbuf, int rcount, \
  61                            MPI_Datatype dtype, MPI_Op op, MPI_Comm comm)
  62
  63 #define COLL_ALLREDUCES(action, COLL_sep) \
  64 COLL_APPLY(action, COLL_ALLREDUCE_SIG, lr) COLL_sep \
  65 COLL_APPLY(action, COLL_ALLREDUCE_SIG, NTS) COLL_sep \
  66 COLL_APPLY(action, COLL_ALLREDUCE_SIG, rab1) COLL_sep \
  67 COLL_APPLY(action, COLL_ALLREDUCE_SIG, rab2) COLL_sep \
  68 COLL_NOTHING(COLL_APPLY(action, COLL_ALLREDUCE_SIG, rab_rdb) COLL_sep) \
  69 COLL_NOTHING(COLL_APPLY(action, COLL_ALLREDUCE_SIG, rab_reduce_scatter) COLL_sep) \
  70 COLL_APPLY(action, COLL_ALLREDUCE_SIG, rab_rsag) COLL_sep \
  71 COLL_APPLY(action, COLL_ALLREDUCE_SIG, rdb) COLL_sep \
  72 COLL_APPLY(action, COLL_ALLREDUCE_SIG, smp_binomial) COLL_sep \
  73 COLL_NOTHING(COLL_APPLY(action, COLL_ALLREDUCE_SIG, smp_binomial_pipeline) COLL_sep) \
  74 COLL_APPLY(action, COLL_ALLREDUCE_SIG, smp_rdb) COLL_sep \
  75 COLL_APPLY(action, COLL_ALLREDUCE_SIG, smp_rsag) COLL_sep \
  76 COLL_APPLY(action, COLL_ALLREDUCE_SIG, smp_rsag_lr) COLL_sep \
  77 COLL_APPLY(action, COLL_ALLREDUCE_SIG, smp_rsag_rab) COLL_sep \
  78 COLL_APPLY(action, COLL_ALLREDUCE_SIG, redbcast)
  79
  80 COLL_ALLREDUCES(COLL_PROTO, COLL_NOsep)
  81
  82
  83 /************
  84  * ALLTOALL *
  85  ************/
  86 #define COLL_ALLTOALL_SIG alltoall, int, \
  87                          (void *send_buff, int send_count, MPI_Datatype send_type, \
  88                           void *recv_buff, int recv_count, MPI_Datatype recv_type, \
  89                           MPI_Comm com)
  90
  91 #define COLL_ALLTOALLS(action, COLL_sep) \
  92 COLL_APPLY(action, COLL_ALLTOALL_SIG, 2dmesh) COLL_sep \
  93 COLL_APPLY(action, COLL_ALLTOALL_SIG, 3dmesh) COLL_sep \
  94 COLL_NOTHING(COLL_APPLY(action, COLL_ALLTOALL_SIG, bruck) COLL_sep) \
  95 COLL_APPLY(action, COLL_ALLTOALL_SIG, pair) COLL_sep \
  96 COLL_APPLY(action, COLL_ALLTOALL_SIG, pair_light_barrier) COLL_sep \
  97 COLL_APPLY(action, COLL_ALLTOALL_SIG, pair_mpi_barrier) COLL_sep \
  98 COLL_APPLY(action, COLL_ALLTOALL_SIG, pair_one_barrier) COLL_sep \
  99 COLL_APPLY(action, COLL_ALLTOALL_SIG, rdb) COLL_sep \
 100 COLL_APPLY(action, COLL_ALLTOALL_SIG, ring) COLL_sep \
 101 COLL_APPLY(action, COLL_ALLTOALL_SIG, ring_light_barrier) COLL_sep \
 102 COLL_APPLY(action, COLL_ALLTOALL_SIG, ring_mpi_barrier) COLL_sep \
 103 COLL_APPLY(action, COLL_ALLTOALL_SIG, ring_one_barrier) COLL_sep \
 104 COLL_APPLY(action, COLL_ALLTOALL_SIG, simple)
 105
 106 COLL_ALLTOALLS(COLL_PROTO, COLL_NOsep)
 107
 108
 109 /*********
 110  * BCAST *
 111  *********/
 112 #define COLL_BCAST_SIG bcast, int, \
 113                       (void *buf, int count, MPI_Datatype datatype, \
 114                        int root, MPI_Comm comm)
 115
 116 #define COLL_BCASTS(action, COLL_sep) \
 117 COLL_APPLY(action, COLL_BCAST_SIG, arrival_nb) COLL_sep \
 118 COLL_APPLY(action, COLL_BCAST_SIG, arrival_pattern_aware) COLL_sep \
 119 COLL_APPLY(action, COLL_BCAST_SIG, arrival_pattern_aware_wait) COLL_sep \
 120 COLL_APPLY(action, COLL_BCAST_SIG, arrival_scatter) COLL_sep \
 121 COLL_APPLY(action, COLL_BCAST_SIG, binomial_tree) COLL_sep \
 122 COLL_APPLY(action, COLL_BCAST_SIG, flattree) COLL_sep \
 123 COLL_APPLY(action, COLL_BCAST_SIG, flattree_pipeline) COLL_sep \
 124 COLL_APPLY(action, COLL_BCAST_SIG, NTSB) COLL_sep \
 125 COLL_APPLY(action, COLL_BCAST_SIG, NTSL) COLL_sep \
 126 COLL_APPLY(action, COLL_BCAST_SIG, NTSL_Isend) COLL_sep \
 127 COLL_APPLY(action, COLL_BCAST_SIG, scatter_LR_allgather) COLL_sep \
 128 COLL_APPLY(action, COLL_BCAST_SIG, scatter_rdb_allgather) COLL_sep \
 129 COLL_APPLY(action, COLL_BCAST_SIG, SMP_binary) COLL_sep \
 130 COLL_APPLY(action, COLL_BCAST_SIG, SMP_binomial) COLL_sep \
 131 COLL_APPLY(action, COLL_BCAST_SIG, SMP_linear) COLL_sep \
 132 COLL_APPLY(action, COLL_BCAST_SIG, TSB)
 133
 134 COLL_BCASTS(COLL_PROTO, COLL_NOsep)
 135
 136
 137 /**********
 138  * REDUCE *
 139  **********/
 140 #define COLL_REDUCE_SIG reduce, int, \
 141                        (void *buf, void *rbuf, int count, MPI_Datatype datatype, \
 142                         MPI_Op op, int root, MPI_Comm comm)
 143
 144 #define COLL_REDUCES(action, COLL_sep) \
 145 COLL_APPLY(action, COLL_REDUCE_SIG, arrival_pattern_aware) COLL_sep \
 146 COLL_APPLY(action, COLL_REDUCE_SIG, binomial) COLL_sep \
 147 COLL_APPLY(action, COLL_REDUCE_SIG, flat_tree) COLL_sep \
 148 COLL_APPLY(action, COLL_REDUCE_SIG, NTSL) COLL_sep \
 149 COLL_APPLY(action, COLL_REDUCE_SIG, scatter_gather)
 150
 151 COLL_REDUCES(COLL_PROTO, COLL_NOsep)
 152
 153 #endif