src/smpi/private.h

   1 /* Copyright (c) 2007, 2009, 2010. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5   * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #ifndef SMPI_PRIVATE_H
   8 #define SMPI_PRIVATE_H
   9
  10 #include "xbt.h"
  11 #include "xbt/xbt_os_time.h"
  12 #include "simix/simix.h"
  13 #include "smpi/smpi.h"
  14 #include "instr/private.h"
  15
  16 struct s_smpi_process_data;
  17 typedef struct s_smpi_process_data *smpi_process_data_t;
  18
  19 #define PERSISTENT     0x0
  20 #define NON_PERSISTENT 0x1
  21 #define SEND           0x0
  22 #define RECV           0x2
  23
  24 typedef struct s_smpi_mpi_request {
  25   void *buf;
  26   size_t size;
  27   int src;
  28   int dst;
  29   int tag;
  30   MPI_Comm comm;
  31   smx_rdv_t rdv;
  32   smx_comm_t pair;
  33   int complete;
  34   MPI_Request match;
  35   unsigned flags;
  36 #ifdef HAVE_TRACING
  37   int send;
  38   int recv;
  39 #endif
  40 } s_smpi_mpi_request_t;
  41
  42 void smpi_process_init(int *argc, char ***argv);
  43 void smpi_process_destroy(void);
  44
  45 smpi_process_data_t smpi_process_data(void);
  46 smpi_process_data_t smpi_process_remote_data(int index);
  47 int smpi_process_count(void);
  48 int smpi_process_index(void);
  49 xbt_os_timer_t smpi_process_timer(void);
  50 void smpi_process_simulated_start(void);
  51 double smpi_process_simulated_elapsed(void);
  52 void print_request(const char *message, MPI_Request request);
  53 void smpi_process_post_send(MPI_Comm comm, MPI_Request request);
  54 void smpi_process_post_recv(MPI_Request request);
  55
  56 void smpi_global_init(void);
  57 void smpi_global_destroy(void);
  58
  59 size_t smpi_datatype_size(MPI_Datatype datatype);
  60 MPI_Aint smpi_datatype_lb(MPI_Datatype datatype);
  61 MPI_Aint smpi_datatype_ub(MPI_Datatype datatype);
  62 int smpi_datatype_extent(MPI_Datatype datatype, MPI_Aint * lb,
  63                          MPI_Aint * extent);
  64 int smpi_datatype_copy(void *sendbuf, int sendcount, MPI_Datatype sendtype,
  65                        void *recvbuf, int recvcount,
  66                        MPI_Datatype recvtype);
  67
  68 MPI_Op smpi_op_new(MPI_User_function * function, int commute);
  69 void smpi_op_destroy(MPI_Op op);
  70 void smpi_op_apply(MPI_Op op, void *invec, void *inoutvec, int *len,
  71                    MPI_Datatype * datatype);
  72
  73 MPI_Group smpi_group_new(int size);
  74 void smpi_group_destroy(MPI_Group group);
  75 void smpi_group_set_mapping(MPI_Group group, int index, int rank);
  76 int smpi_group_index(MPI_Group group, int rank);
  77 int smpi_group_rank(MPI_Group group, int index);
  78 int smpi_group_use(MPI_Group group);
  79 int smpi_group_unuse(MPI_Group group);
  80 int smpi_group_size(MPI_Group group);
  81 int smpi_group_compare(MPI_Group group1, MPI_Group group2);
  82
  83 MPI_Comm smpi_comm_new(MPI_Group group);
  84 void smpi_comm_destroy(MPI_Comm comm);
  85 MPI_Group smpi_comm_group(MPI_Comm comm);
  86 int smpi_comm_size(MPI_Comm comm);
  87 int smpi_comm_rank(MPI_Comm comm);
  88
  89 MPI_Request smpi_mpi_send_init(void *buf, int count, MPI_Datatype datatype,
  90                                int dst, int tag, MPI_Comm comm);
  91 MPI_Request smpi_mpi_recv_init(void *buf, int count, MPI_Datatype datatype,
  92                                int src, int tag, MPI_Comm comm);
  93 void smpi_mpi_start(MPI_Request request);
  94 void smpi_mpi_startall(int count, MPI_Request * requests);
  95 void smpi_mpi_request_free(MPI_Request * request);
  96 MPI_Request smpi_isend_init(void *buf, int count, MPI_Datatype datatype,
  97                             int dst, int tag, MPI_Comm comm);
  98 MPI_Request smpi_mpi_isend(void *buf, int count, MPI_Datatype datatype,
  99                            int dst, int tag, MPI_Comm comm);
 100 MPI_Request smpi_irecv_init(void *buf, int count, MPI_Datatype datatype,
 101                             int src, int tag, MPI_Comm comm);
 102 MPI_Request smpi_mpi_irecv(void *buf, int count, MPI_Datatype datatype,
 103                            int src, int tag, MPI_Comm comm);
 104 void smpi_mpi_recv(void *buf, int count, MPI_Datatype datatype, int src,
 105                    int tag, MPI_Comm comm, MPI_Status * status);
 106 void smpi_mpi_send(void *buf, int count, MPI_Datatype datatype, int dst,
 107                    int tag, MPI_Comm comm);
 108 void smpi_mpi_sendrecv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
 109                        int dst, int sendtag, void *recvbuf, int recvcount,
 110                        MPI_Datatype recvtype, int src, int recvtag,
 111                        MPI_Comm comm, MPI_Status * status);
 112 int smpi_mpi_test(MPI_Request * request, MPI_Status * status);
 113 int smpi_mpi_testany(int count, MPI_Request requests[], int *index,
 114                      MPI_Status * status);
 115 int smpi_mpi_get_count(MPI_Status * status, MPI_Datatype datatype);
 116 void smpi_mpi_wait(MPI_Request * request, MPI_Status * status);
 117 int smpi_mpi_waitany(int count, MPI_Request requests[],
 118                      MPI_Status * status);
 119 void smpi_mpi_waitall(int count, MPI_Request requests[],
 120                       MPI_Status status[]);
 121 int smpi_mpi_waitsome(int incount, MPI_Request requests[], int *indices,
 122                       MPI_Status status[]);
 123 void smpi_mpi_bcast(void *buf, int count, MPI_Datatype datatype, int root,
 124                     MPI_Comm comm);
 125 void smpi_mpi_barrier(MPI_Comm comm);
 126 void smpi_mpi_gather(void *sendbuf, int sendcount, MPI_Datatype sendtype,
 127                      void *recvbuf, int recvcount, MPI_Datatype recvtype,
 128                      int root, MPI_Comm comm);
 129 void smpi_mpi_gatherv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
 130                       void *recvbuf, int *recvcounts, int *displs,
 131                       MPI_Datatype recvtype, int root, MPI_Comm comm);
 132 void smpi_mpi_allgather(void *sendbuf, int sendcount,
 133                         MPI_Datatype sendtype, void *recvbuf,
 134                         int recvcount, MPI_Datatype recvtype,
 135                         MPI_Comm comm);
 136 void smpi_mpi_allgatherv(void *sendbuf, int sendcount,
 137                          MPI_Datatype sendtype, void *recvbuf,
 138                          int *recvcounts, int *displs,
 139                          MPI_Datatype recvtype, MPI_Comm comm);
 140 void smpi_mpi_scatter(void *sendbuf, int sendcount, MPI_Datatype sendtype,
 141                       void *recvbuf, int recvcount, MPI_Datatype recvtype,
 142                       int root, MPI_Comm comm);
 143 void smpi_mpi_scatterv(void *sendbuf, int *sendcounts, int *displs,
 144                        MPI_Datatype sendtype, void *recvbuf, int recvcount,
 145                        MPI_Datatype recvtype, int root, MPI_Comm comm);
 146 void smpi_mpi_reduce(void *sendbuf, void *recvbuf, int count,
 147                      MPI_Datatype datatype, MPI_Op op, int root,
 148                      MPI_Comm comm);
 149 void smpi_mpi_allreduce(void *sendbuf, void *recvbuf, int count,
 150                         MPI_Datatype datatype, MPI_Op op, MPI_Comm comm);
 151 void smpi_mpi_scan(void *sendbuf, void *recvbuf, int count,
 152                    MPI_Datatype datatype, MPI_Op op, MPI_Comm comm);
 153
 154 void nary_tree_bcast(void *buf, int count, MPI_Datatype datatype, int root,
 155                      MPI_Comm comm, int arity);
 156 void nary_tree_barrier(MPI_Comm comm, int arity);
 157
 158 int smpi_coll_tuned_alltoall_bruck(void *sendbuf, int sendcount,
 159                                    MPI_Datatype sendtype, void *recvbuf,
 160                                    int recvcount, MPI_Datatype recvtype,
 161                                    MPI_Comm comm);
 162 int smpi_coll_tuned_alltoall_basic_linear(void *sendbuf, int sendcount,
 163                                           MPI_Datatype sendtype,
 164                                           void *recvbuf, int recvcount,
 165                                           MPI_Datatype recvtype,
 166                                           MPI_Comm comm);
 167 int smpi_coll_tuned_alltoall_pairwise(void *sendbuf, int sendcount,
 168                                       MPI_Datatype sendtype, void *recvbuf,
 169                                       int recvcount, MPI_Datatype recvtype,
 170                                       MPI_Comm comm);
 171 int smpi_coll_basic_alltoallv(void *sendbuf, int *sendcounts,
 172                               int *senddisps, MPI_Datatype sendtype,
 173                               void *recvbuf, int *recvcounts,
 174                               int *recvdisps, MPI_Datatype recvtype,
 175                               MPI_Comm comm);
 176
 177 // utilities
 178 void smpi_bench_destroy(void);
 179 void smpi_bench_begin(int rank, const char *mpi_call);
 180 void smpi_bench_end(int rank, const char *mpi_call);
 181
 182 #endif