1 /* Copyright (c) 2010. The SimGrid Team.
2 * All rights reserved. */
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. */
16 static xbt_dynar_t comm_lookup = NULL;
17 static xbt_dict_t request_lookup = NULL;
18 static xbt_dynar_t datatype_lookup = NULL;
19 static xbt_dynar_t op_lookup = NULL;
21 #define KEY_SIZE (sizeof(int) * 2 + 1)
23 static int new_comm(MPI_Comm comm) {
24 xbt_dynar_push(comm_lookup, &comm);
25 return (int)xbt_dynar_length(comm_lookup) - 1;
28 static MPI_Comm get_comm(int comm) {
31 } else if(comm_lookup && comm >= 0 && comm < (int)xbt_dynar_length(comm_lookup)) {
32 return *(MPI_Comm*)xbt_dynar_get_ptr(comm_lookup, comm);
37 static char* get_key(char* key, int id) {
38 snprintf(key, KEY_SIZE, "%x", id);
42 static int new_request(MPI_Request req) {
43 static int request_id = INT_MIN;
46 xbt_dict_set(request_lookup, get_key(key, request_id), req, NULL);
50 static MPI_Request find_request(int req) {
53 return (MPI_Request)xbt_dict_get(request_lookup, get_key(key, req));
56 static int new_datatype(MPI_Datatype datatype) {
57 xbt_dynar_push(datatype_lookup, &datatype);
58 return (int)xbt_dynar_length(datatype_lookup) - 1;
61 static MPI_Datatype get_datatype(int datatype) {
63 ? *(MPI_Datatype*)xbt_dynar_get_ptr(datatype_lookup, datatype)
67 static int new_op(MPI_Op op) {
68 xbt_dynar_push(op_lookup, &op);
69 return (int)xbt_dynar_length(op_lookup) - 1;
72 static MPI_Op get_op(int op) {
74 ? *(MPI_Op*)xbt_dynar_get_ptr(op_lookup, op)
78 void mpi_init__(int* ierr) {
79 comm_lookup = xbt_dynar_new(sizeof(MPI_Comm), NULL);
80 new_comm(MPI_COMM_WORLD);
82 request_lookup = xbt_dict_new_homogeneous(NULL);
84 datatype_lookup = xbt_dynar_new(sizeof(MPI_Datatype), NULL);
85 new_datatype(MPI_BYTE);
86 new_datatype(MPI_CHAR);
87 new_datatype(MPI_INT);
88 new_datatype(MPI_INT);
89 new_datatype(MPI_INT8_T);
90 new_datatype(MPI_INT16_T);
91 new_datatype(MPI_INT32_T);
92 new_datatype(MPI_INT64_T);
93 new_datatype(MPI_FLOAT);
94 new_datatype(MPI_FLOAT);
95 new_datatype(MPI_DOUBLE);
96 new_datatype(MPI_DOUBLE);
97 new_datatype(MPI_C_FLOAT_COMPLEX);
98 new_datatype(MPI_C_DOUBLE_COMPLEX);
99 new_datatype(MPI_2INT);
100 new_datatype(MPI_UINT8_T);
101 new_datatype(MPI_UINT16_T);
102 new_datatype(MPI_UINT32_T);
103 new_datatype(MPI_UINT64_T);
104 new_datatype(MPI_2FLOAT);
105 new_datatype(MPI_2DOUBLE);
108 op_lookup = xbt_dynar_new(sizeof(MPI_Op), NULL);
122 /* smpif2c is responsible for generating a call with the final arguments */
123 *ierr = MPI_Init(NULL, NULL);
126 void mpi_finalize__(int* ierr) {
127 *ierr = MPI_Finalize();
128 xbt_dynar_free(&op_lookup);
130 xbt_dynar_free(&datatype_lookup);
131 datatype_lookup = NULL;
132 xbt_dict_free(&request_lookup);
133 request_lookup = NULL;
134 xbt_dynar_free(&comm_lookup);
138 void mpi_abort__(int* comm, int* errorcode, int* ierr) {
139 *ierr = MPI_Abort(get_comm(*comm), *errorcode);
142 void mpi_comm_rank__(int* comm, int* rank, int* ierr) {
143 *ierr = MPI_Comm_rank(get_comm(*comm), rank);
146 void mpi_comm_size__(int* comm, int* size, int* ierr) {
147 *ierr = MPI_Comm_size(get_comm(*comm), size);
150 double mpi_wtime__(void) {
154 double mpi_wtick__(void) {
158 void mpi_comm_dup__(int* comm, int* newcomm, int* ierr) {
161 *ierr = MPI_Comm_dup(get_comm(*comm), &tmp);
162 if(*ierr == MPI_SUCCESS) {
163 *newcomm = new_comm(tmp);
167 void mpi_comm_split__(int* comm, int* color, int* key, int* comm_out, int* ierr) {
170 *ierr = MPI_Comm_split(get_comm(*comm), *color, *key, &tmp);
171 if(*ierr == MPI_SUCCESS) {
172 *comm_out = new_comm(tmp);
176 void mpi_send_init__(void *buf, int* count, int* datatype, int* dst, int* tag,
177 int* comm, int* request, int* ierr) {
180 *ierr = MPI_Send_init(buf, *count, get_datatype(*datatype), *dst, *tag,
181 get_comm(*comm), &req);
182 if(*ierr == MPI_SUCCESS) {
183 *request = new_request(req);
187 void mpi_isend__(void *buf, int* count, int* datatype, int* dst,
188 int* tag, int* comm, int* request, int* ierr) {
191 *ierr = MPI_Isend(buf, *count, get_datatype(*datatype), *dst, *tag,
192 get_comm(*comm), &req);
193 if(*ierr == MPI_SUCCESS) {
194 *request = new_request(req);
198 void mpi_send__(void* buf, int* count, int* datatype, int* dst,
199 int* tag, int* comm, int* ierr) {
200 *ierr = MPI_Send(buf, *count, get_datatype(*datatype), *dst, *tag,
204 void mpi_sendrecv__(void* sendbuf, int* sendcount, int* sendtype, int* dst,
205 int* sendtag, void *recvbuf, int* recvcount,
206 int* recvtype, int* src, int* recvtag,
207 int* comm, MPI_Status* status, int* ierr) {
208 *ierr = MPI_Sendrecv(sendbuf, *sendcount, get_datatype(*sendtype), *dst,
209 *sendtag, recvbuf, *recvcount,get_datatype(*recvtype), *src, *recvtag,
210 get_comm(*comm), status);
213 void mpi_recv_init__(void *buf, int* count, int* datatype, int* src, int* tag,
214 int* comm, int* request, int* ierr) {
217 *ierr = MPI_Recv_init(buf, *count, get_datatype(*datatype), *src, *tag,
218 get_comm(*comm), &req);
219 if(*ierr == MPI_SUCCESS) {
220 *request = new_request(req);
224 void mpi_irecv__(void *buf, int* count, int* datatype, int* src, int* tag,
225 int* comm, int* request, int* ierr) {
228 *ierr = MPI_Irecv(buf, *count, get_datatype(*datatype), *src, *tag,
229 get_comm(*comm), &req);
230 if(*ierr == MPI_SUCCESS) {
231 *request = new_request(req);
235 void mpi_recv__(void* buf, int* count, int* datatype, int* src,
236 int* tag, int* comm, MPI_Status* status, int* ierr) {
237 *ierr = MPI_Recv(buf, *count, get_datatype(*datatype), *src, *tag,
238 get_comm(*comm), status);
241 void mpi_start__(int* request, int* ierr) {
242 MPI_Request req = find_request(*request);
244 *ierr = MPI_Start(&req);
247 void mpi_startall__(int* count, int* requests, int* ierr) {
251 reqs = xbt_new(MPI_Request, *count);
252 for(i = 0; i < *count; i++) {
253 reqs[i] = find_request(requests[i]);
255 *ierr = MPI_Startall(*count, reqs);
259 void mpi_wait__(int* request, MPI_Status* status, int* ierr) {
260 MPI_Request req = find_request(*request);
262 *ierr = MPI_Wait(&req, status);
265 void mpi_waitany__(int* count, int* requests, int* index, MPI_Status* status, int* ierr) {
269 reqs = xbt_new(MPI_Request, *count);
270 for(i = 0; i < *count; i++) {
271 reqs[i] = find_request(requests[i]);
273 *ierr = MPI_Waitany(*count, reqs, index, status);
277 void mpi_waitall__(int* count, int* requests, MPI_Status* status, int* ierr) {
281 reqs = xbt_new(MPI_Request, *count);
282 for(i = 0; i < *count; i++) {
283 reqs[i] = find_request(requests[i]);
285 *ierr = MPI_Waitall(*count, reqs, status);
289 void mpi_barrier__(int* comm, int* ierr) {
290 *ierr = MPI_Barrier(get_comm(*comm));
293 void mpi_bcast__(void *buf, int* count, int* datatype, int* root, int* comm, int* ierr) {
294 *ierr = MPI_Bcast(buf, *count, get_datatype(*datatype), *root, get_comm(*comm));
297 void mpi_reduce__(void* sendbuf, void* recvbuf, int* count,
298 int* datatype, int* op, int* root, int* comm, int* ierr) {
299 *ierr = MPI_Reduce(sendbuf, recvbuf, *count,
300 get_datatype(*datatype), get_op(*op), *root, get_comm(*comm));
303 void mpi_allreduce__(void* sendbuf, void* recvbuf, int* count, int* datatype,
304 int* op, int* comm, int* ierr) {
305 *ierr = MPI_Allreduce(sendbuf, recvbuf, *count, get_datatype(*datatype),
306 get_op(*op), get_comm(*comm));
309 void mpi_scatter__(void* sendbuf, int* sendcount, int* sendtype,
310 void* recvbuf, int* recvcount, int* recvtype,
311 int* root, int* comm, int* ierr) {
312 *ierr = MPI_Scatter(sendbuf, *sendcount, get_datatype(*sendtype),
313 recvbuf, *recvcount, get_datatype(*recvtype), *root, get_comm(*comm));
316 void mpi_gather__(void* sendbuf, int* sendcount, int* sendtype,
317 void* recvbuf, int* recvcount, int* recvtype,
318 int* root, int* comm, int* ierr) {
319 *ierr = MPI_Gather(sendbuf, *sendcount, get_datatype(*sendtype),
320 recvbuf, *recvcount, get_datatype(*recvtype), *root, get_comm(*comm));
323 void mpi_gatherv__(void* sendbuf, int* sendcount, int* sendtype,
324 void* recvbuf, int* recvcounts, int* displs, int* recvtype,
325 int* root, int* comm, int* ierr) {
326 *ierr = MPI_Gatherv(sendbuf, *sendcount, get_datatype(*sendtype),
327 recvbuf, recvcounts, displs, get_datatype(*recvtype), *root, get_comm(*comm));
330 void mpi_allgather__(void* sendbuf, int* sendcount, int* sendtype,
331 void* recvbuf, int* recvcount, int* recvtype,
332 int* comm, int* ierr) {
333 *ierr = MPI_Allgather(sendbuf, *sendcount, get_datatype(*sendtype),
334 recvbuf, *recvcount, get_datatype(*recvtype), get_comm(*comm));
337 void mpi_allgatherv__(void* sendbuf, int* sendcount, int* sendtype,
338 void* recvbuf, int* recvcount,int* displs, int* recvtype,
339 int* comm, int* ierr) {
340 *ierr = MPI_Allgatherv(sendbuf, *sendcount, get_datatype(*sendtype),
341 recvbuf, recvcount, displs, get_datatype(*recvtype), get_comm(*comm));
344 void mpi_scan__(void* sendbuf, void* recvbuf, int* count, int* datatype,
345 int* op, int* comm, int* ierr) {
346 *ierr = MPI_Scan(sendbuf, recvbuf, *count, get_datatype(*datatype),
347 get_op(*op), get_comm(*comm));
350 void mpi_alltoall__(void* sendbuf, int* sendcount, int* sendtype,
351 void* recvbuf, int* recvcount, int* recvtype, int* comm, int* ierr) {
352 *ierr = MPI_Alltoall(sendbuf, *sendcount, get_datatype(*sendtype),
353 recvbuf, *recvcount, get_datatype(*recvtype), get_comm(*comm));
356 void mpi_test__ (int * request, int *flag, MPI_Status * status, int* ierr){
357 MPI_Request req = find_request(*request);
358 *ierr= MPI_Test(&req, flag, status);
360 void mpi_get_processor_name__(char *name, int *resultlen, int* ierr){
361 *ierr = MPI_Get_processor_name(name, resultlen);
364 void mpi_get_count__(MPI_Status * status, int* datatype, int *count, int* ierr){
365 *ierr = MPI_Get_count(status, get_datatype(*datatype), count);
368 void mpi_type_extent__(int* datatype, MPI_Aint * extent, int* ierr){
369 *ierr= MPI_Type_extent(get_datatype(*datatype), extent);
372 void mpi_type_ub__(int* datatype, MPI_Aint * disp, int* ierr){
373 *ierr= MPI_Type_ub(get_datatype(*datatype), disp);
376 void mpi_type_lb__(int* datatype, MPI_Aint * extent, int* ierr){
377 *ierr= MPI_Type_extent(get_datatype(*datatype), extent);
380 void mpi_type_size__(int* datatype, int *size, int* ierr)
382 *ierr = MPI_Type_size(get_datatype(*datatype), size);