Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
add an "smpi/async_small_thres" configure option to smpi, to allow to toy with the...
[simgrid.git] / 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 "simgrid/simix.h"
13 #include "smpi/smpi.h"
14 #include "smpi/smpif.h"
15 #include "smpi/smpi_cocci.h"
16 #include "instr/instr_private.h"
17
18 struct s_smpi_process_data;
19 typedef struct s_smpi_process_data *smpi_process_data_t;
20
21 #define PERSISTENT     0x1
22 #define NON_PERSISTENT 0x2
23 #define SEND           0x4
24 #define RECV           0x8
25
26 typedef struct s_smpi_mpi_request {
27   void *buf;
28   size_t size;
29   int src;
30   int dst;
31   int tag;
32   MPI_Comm comm;
33   smx_action_t action;
34   unsigned flags;
35 #ifdef HAVE_TRACING
36   int send;
37   int recv;
38 #endif
39 } s_smpi_mpi_request_t;
40
41 void smpi_process_init(int *argc, char ***argv);
42 void smpi_process_destroy(void);
43 void smpi_process_finalize(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 smx_rdv_t smpi_process_mailbox(void);
49 smx_rdv_t smpi_process_remote_mailbox(int index);
50 smx_rdv_t smpi_process_mailbox_small(void);
51 smx_rdv_t smpi_process_remote_mailbox_small(int index);
52 xbt_os_timer_t smpi_process_timer(void);
53 void smpi_process_simulated_start(void);
54 double smpi_process_simulated_elapsed(void);
55
56 void print_request(const char *message, MPI_Request request);
57
58 void smpi_global_init(void);
59 void smpi_global_destroy(void);
60
61 size_t smpi_datatype_size(MPI_Datatype datatype);
62 MPI_Aint smpi_datatype_lb(MPI_Datatype datatype);
63 MPI_Aint smpi_datatype_ub(MPI_Datatype datatype);
64 int smpi_datatype_extent(MPI_Datatype datatype, MPI_Aint * lb,
65                          MPI_Aint * extent);
66 int smpi_datatype_copy(void *sendbuf, int sendcount, MPI_Datatype sendtype,
67                        void *recvbuf, int recvcount,
68                        MPI_Datatype recvtype);
69
70 MPI_Op smpi_op_new(MPI_User_function * function, int commute);
71 void smpi_op_destroy(MPI_Op op);
72 void smpi_op_apply(MPI_Op op, void *invec, void *inoutvec, int *len,
73                    MPI_Datatype * datatype);
74
75 MPI_Group smpi_group_new(int size);
76 void smpi_group_destroy(MPI_Group group);
77 void smpi_group_set_mapping(MPI_Group group, int index, int rank);
78 int smpi_group_index(MPI_Group group, int rank);
79 int smpi_group_rank(MPI_Group group, int index);
80 int smpi_group_use(MPI_Group group);
81 int smpi_group_unuse(MPI_Group group);
82 int smpi_group_size(MPI_Group group);
83 int smpi_group_compare(MPI_Group group1, MPI_Group group2);
84
85 MPI_Comm smpi_comm_new(MPI_Group group);
86 void smpi_comm_destroy(MPI_Comm comm);
87 MPI_Group smpi_comm_group(MPI_Comm comm);
88 int smpi_comm_size(MPI_Comm comm);
89 void smpi_comm_get_name(MPI_Comm comm, char* name, int* len);
90 int smpi_comm_rank(MPI_Comm comm);
91 MPI_Comm smpi_comm_split(MPI_Comm comm, int color, int key);
92
93 MPI_Request smpi_mpi_send_init(void *buf, int count, MPI_Datatype datatype,
94                                int dst, int tag, MPI_Comm comm);
95 MPI_Request smpi_mpi_recv_init(void *buf, int count, MPI_Datatype datatype,
96                                int src, int tag, MPI_Comm comm);
97 void smpi_mpi_start(MPI_Request request);
98 void smpi_mpi_startall(int count, MPI_Request * requests);
99 void smpi_mpi_request_free(MPI_Request * request);
100 MPI_Request smpi_isend_init(void *buf, int count, MPI_Datatype datatype,
101                             int dst, int tag, MPI_Comm comm);
102 MPI_Request smpi_mpi_isend(void *buf, int count, MPI_Datatype datatype,
103                            int dst, int tag, MPI_Comm comm);
104 MPI_Request smpi_irecv_init(void *buf, int count, MPI_Datatype datatype,
105                             int src, int tag, MPI_Comm comm);
106 MPI_Request smpi_mpi_irecv(void *buf, int count, MPI_Datatype datatype,
107                            int src, int tag, MPI_Comm comm);
108 void smpi_mpi_recv(void *buf, int count, MPI_Datatype datatype, int src,
109                    int tag, MPI_Comm comm, MPI_Status * status);
110 void smpi_mpi_send(void *buf, int count, MPI_Datatype datatype, int dst,
111                    int tag, MPI_Comm comm);
112 void smpi_mpi_sendrecv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
113                        int dst, int sendtag, void *recvbuf, int recvcount,
114                        MPI_Datatype recvtype, int src, int recvtag,
115                        MPI_Comm comm, MPI_Status * status);
116 int smpi_mpi_test(MPI_Request * request, MPI_Status * status);
117 int smpi_mpi_testany(int count, MPI_Request requests[], int *index,
118                      MPI_Status * status);
119 int smpi_mpi_get_count(MPI_Status * status, MPI_Datatype datatype);
120 void smpi_mpi_wait(MPI_Request * request, MPI_Status * status);
121 int smpi_mpi_waitany(int count, MPI_Request requests[],
122                      MPI_Status * status);
123 void smpi_mpi_waitall(int count, MPI_Request requests[],
124                       MPI_Status status[]);
125 int smpi_mpi_waitsome(int incount, MPI_Request requests[], int *indices,
126                       MPI_Status status[]);
127 void smpi_mpi_bcast(void *buf, int count, MPI_Datatype datatype, int root,
128                     MPI_Comm comm);
129 void smpi_mpi_barrier(MPI_Comm comm);
130 void smpi_mpi_gather(void *sendbuf, int sendcount, MPI_Datatype sendtype,
131                      void *recvbuf, int recvcount, MPI_Datatype recvtype,
132                      int root, MPI_Comm comm);
133 void smpi_mpi_gatherv(void *sendbuf, int sendcount, MPI_Datatype sendtype,
134                       void *recvbuf, int *recvcounts, int *displs,
135                       MPI_Datatype recvtype, int root, MPI_Comm comm);
136 void smpi_mpi_allgather(void *sendbuf, int sendcount,
137                         MPI_Datatype sendtype, void *recvbuf,
138                         int recvcount, MPI_Datatype recvtype,
139                         MPI_Comm comm);
140 void smpi_mpi_allgatherv(void *sendbuf, int sendcount,
141                          MPI_Datatype sendtype, void *recvbuf,
142                          int *recvcounts, int *displs,
143                          MPI_Datatype recvtype, MPI_Comm comm);
144 void smpi_mpi_scatter(void *sendbuf, int sendcount, MPI_Datatype sendtype,
145                       void *recvbuf, int recvcount, MPI_Datatype recvtype,
146                       int root, MPI_Comm comm);
147 void smpi_mpi_scatterv(void *sendbuf, int *sendcounts, int *displs,
148                        MPI_Datatype sendtype, void *recvbuf, int recvcount,
149                        MPI_Datatype recvtype, int root, MPI_Comm comm);
150 void smpi_mpi_reduce(void *sendbuf, void *recvbuf, int count,
151                      MPI_Datatype datatype, MPI_Op op, int root,
152                      MPI_Comm comm);
153 void smpi_mpi_allreduce(void *sendbuf, void *recvbuf, int count,
154                         MPI_Datatype datatype, MPI_Op op, MPI_Comm comm);
155 void smpi_mpi_scan(void *sendbuf, void *recvbuf, int count,
156                    MPI_Datatype datatype, MPI_Op op, MPI_Comm comm);
157
158 void nary_tree_bcast(void *buf, int count, MPI_Datatype datatype, int root,
159                      MPI_Comm comm, int arity);
160 void nary_tree_barrier(MPI_Comm comm, int arity);
161
162 int smpi_coll_tuned_alltoall_bruck(void *sendbuf, int sendcount,
163                                    MPI_Datatype sendtype, void *recvbuf,
164                                    int recvcount, MPI_Datatype recvtype,
165                                    MPI_Comm comm);
166 int smpi_coll_tuned_alltoall_basic_linear(void *sendbuf, int sendcount,
167                                           MPI_Datatype sendtype,
168                                           void *recvbuf, int recvcount,
169                                           MPI_Datatype recvtype,
170                                           MPI_Comm comm);
171 int smpi_coll_tuned_alltoall_pairwise(void *sendbuf, int sendcount,
172                                       MPI_Datatype sendtype, void *recvbuf,
173                                       int recvcount, MPI_Datatype recvtype,
174                                       MPI_Comm comm);
175 int smpi_coll_basic_alltoallv(void *sendbuf, int *sendcounts,
176                               int *senddisps, MPI_Datatype sendtype,
177                               void *recvbuf, int *recvcounts,
178                               int *recvdisps, MPI_Datatype recvtype,
179                               MPI_Comm comm);
180
181 // utilities
182 void smpi_bench_destroy(void);
183 void smpi_bench_begin(void);
184 void smpi_bench_end(void);
185 void smpi_execute_flops(double flops);
186
187 // f77 wrappers
188 void mpi_init__(int*);
189 void mpi_finalize__(int*);
190 void mpi_abort__(int* comm, int* errorcode, int* ierr);
191 void mpi_comm_rank__(int* comm, int* rank, int* ierr);
192 void mpi_comm_size__(int* comm, int* size, int* ierr);
193 double mpi_wtime__(void);
194
195 void mpi_comm_dup__(int* comm, int* newcomm, int* ierr);
196 void mpi_comm_split__(int* comm, int* color, int* key, int* comm_out, int* ierr);
197
198 void mpi_send_init__(void *buf, int* count, int* datatype, int* dst, int* tag,
199                      int* comm, int* request, int* ierr);
200 void mpi_isend__(void *buf, int* count, int* datatype, int* dst,
201                  int* tag, int* comm, int* request, int* ierr);
202 void mpi_send__(void* buf, int* count, int* datatype, int* dst,
203                 int* tag, int* comm, int* ierr);
204 void mpi_recv_init__(void *buf, int* count, int* datatype, int* src, int* tag,
205                      int* comm, int* request, int* ierr);
206 void mpi_irecv__(void *buf, int* count, int* datatype, int* src, int* tag,
207                  int* comm, int* request, int* ierr);
208 void mpi_recv__(void* buf, int* count, int* datatype, int* src,
209                 int* tag, int* comm, MPI_Status* status, int* ierr);
210 void mpi_start__(int* request, int* ierr);
211 void mpi_startall__(int* count, int* requests, int* ierr);
212 void mpi_wait__(int* request, MPI_Status* status, int* ierr);
213 void mpi_waitany__(int* count, int* requests, int* index, MPI_Status* status, int* ierr);
214 void mpi_waitall__(int* count, int* requests, MPI_Status* status, int* ierr);
215
216 void mpi_barrier__(int* comm, int* ierr);
217 void mpi_bcast__(void* buf, int* count, int* datatype, int* root, int* comm, int* ierr);
218 void mpi_reduce__(void* sendbuf, void* recvbuf, int* count,
219                   int* datatype, int* op, int* root, int* comm, int* ierr);
220 void mpi_allreduce__(void* sendbuf, void* recvbuf, int* count, int* datatype,
221                      int* op, int* comm, int* ierr);
222 void mpi_scatter__(void* sendbuf, int* sendcount, int* sendtype,
223                    void* recvbuf, int* recvcount, int* recvtype, 
224                    int* root, int* comm, int* ierr);
225 void mpi_gather__(void* sendbuf, int* sendcount, int* sendtype,
226                   void* recvbuf, int* recvcount, int* recvtype,
227                   int* root, int* comm, int* ierr);
228 void mpi_allgather__(void* sendbuf, int* sendcount, int* sendtype,
229                      void* recvbuf, int* recvcount, int* recvtype,
230                      int* comm, int* ierr);
231 void mpi_scan__(void* sendbuf, void* recvbuf, int* count, int* datatype,
232                 int* op, int* comm, int* ierr);
233 void mpi_alltoall__(void* sendbuf, int* sendcount, int* sendtype,
234                     void* recvbuf, int* recvcount, int* recvtype, int* comm, int* ierr);
235
236 #endif