Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Support more MPI calls in fortran.
[simgrid.git] / src / smpi / smpi_f77.c
1 /* Copyright (c) 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 #include <limits.h>
8 #include <stdio.h>
9
10 #include "private.h"
11 #include "xbt.h"
12
13 extern int xargc;
14 extern char** xargv;
15
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;
20
21 #define KEY_SIZE (sizeof(int) * 2 + 1)
22
23 static int new_comm(MPI_Comm comm) {
24   xbt_dynar_push(comm_lookup, &comm);
25   return (int)xbt_dynar_length(comm_lookup) - 1;
26 }
27
28 static MPI_Comm get_comm(int comm) {
29   if(comm == -2) {
30     return MPI_COMM_SELF;
31   } else if(comm >= 0) {
32     return *(MPI_Comm*)xbt_dynar_get_ptr(comm_lookup, comm);
33   }
34   return MPI_COMM_NULL;
35 }
36
37 static char* get_key(char* key, int id) {
38   snprintf(key, KEY_SIZE, "%x", id);
39   return key;
40 }
41
42 static int new_request(MPI_Request req) {
43   static int request_id = INT_MIN;
44   char key[KEY_SIZE];
45
46   xbt_dict_set(request_lookup, get_key(key, request_id), req, NULL);
47   return request_id++;
48 }
49
50 static MPI_Request find_request(int req) {
51   char key[KEY_SIZE];
52    
53   return (MPI_Request)xbt_dict_get(request_lookup, get_key(key, req));
54 }
55
56 static int new_datatype(MPI_Datatype datatype) {
57   xbt_dynar_push(datatype_lookup, &datatype);
58   return (int)xbt_dynar_length(datatype_lookup) - 1;
59 }
60
61 static MPI_Datatype get_datatype(int datatype) {
62   return datatype >= 0
63          ? *(MPI_Datatype*)xbt_dynar_get_ptr(datatype_lookup, datatype)
64          : MPI_DATATYPE_NULL;
65 }
66
67 static int new_op(MPI_Op op) {
68   xbt_dynar_push(op_lookup, &op);
69   return (int)xbt_dynar_length(op_lookup) - 1;
70 }
71
72 static MPI_Op get_op(int op) {
73    return op >= 0
74           ? *(MPI_Op*)xbt_dynar_get_ptr(op_lookup, op)
75           : MPI_OP_NULL;
76 }
77
78 void mpi_init__(int* ierr) {
79    comm_lookup = xbt_dynar_new(sizeof(MPI_Comm), NULL);
80    new_comm(MPI_COMM_WORLD);
81
82    request_lookup = xbt_dict_new();
83
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
105    op_lookup = xbt_dynar_new(sizeof(MPI_Op), NULL);
106    new_op(MPI_MAX);
107    new_op(MPI_MIN);
108    new_op(MPI_MAXLOC);
109    new_op(MPI_MINLOC);
110    new_op(MPI_SUM);
111    new_op(MPI_PROD);
112    new_op(MPI_LAND);
113    new_op(MPI_LOR);
114    new_op(MPI_LXOR);
115    new_op(MPI_BAND);
116    new_op(MPI_BOR);
117    new_op(MPI_BXOR);
118
119    /* smpif2c is responsible for generating a call with the final arguments */
120    *ierr = MPI_Init(NULL, NULL);
121 }
122
123 void mpi_finalize__(int* ierr) {
124    *ierr = MPI_Finalize();
125    xbt_dynar_free(&op_lookup);
126    xbt_dynar_free(&datatype_lookup);
127    xbt_dict_free(&request_lookup);
128    xbt_dynar_free(&comm_lookup);
129 }
130
131 void mpi_abort__(int* comm, int* errorcode, int* ierr) {
132   *ierr = MPI_Abort(get_comm(*comm), *errorcode);
133 }
134
135 void mpi_comm_rank__(int* comm, int* rank, int* ierr) {
136    *ierr = MPI_Comm_rank(get_comm(*comm), rank);
137 }
138
139 void mpi_comm_size__(int* comm, int* size, int* ierr) {
140    *ierr = MPI_Comm_size(get_comm(*comm), size);
141 }
142
143 double mpi_wtime__(void) {
144    return MPI_Wtime();
145 }
146
147 void mpi_comm_dup__(int* comm, int* newcomm, int* ierr) {
148   MPI_Comm tmp;
149
150   *ierr = MPI_Comm_dup(get_comm(*comm), &tmp);
151   if(*ierr == MPI_SUCCESS) {
152     *newcomm = new_comm(tmp);
153   }
154 }
155
156 void mpi_comm_split__(int* comm, int* color, int* key, int* comm_out, int* ierr) {
157   MPI_Comm tmp;
158
159   *ierr = MPI_Comm_split(get_comm(*comm), *color, *key, &tmp);
160   if(*ierr == MPI_SUCCESS) {
161     *comm_out = new_comm(tmp);
162   }
163 }
164
165 void mpi_send_init__(void *buf, int* count, int* datatype, int* dst, int* tag,
166                      int* comm, int* request, int* ierr) {
167   MPI_Request req;
168
169   *ierr = MPI_Send_init(buf, *count, get_datatype(*datatype), *dst, *tag,
170                         get_comm(*comm), &req);
171   if(*ierr == MPI_SUCCESS) {
172     *request = new_request(req);
173   }
174 }
175
176 void mpi_isend__(void *buf, int* count, int* datatype, int* dst,
177                  int* tag, int* comm, int* request, int* ierr) {
178   MPI_Request req;
179
180   *ierr = MPI_Isend(buf, *count, get_datatype(*datatype), *dst, *tag,
181                     get_comm(*comm), &req);
182   if(*ierr == MPI_SUCCESS) {
183     *request = new_request(req);
184   }
185 }
186
187 void mpi_send__(void* buf, int* count, int* datatype, int* dst,
188                 int* tag, int* comm, int* ierr) {
189    *ierr = MPI_Send(buf, *count, get_datatype(*datatype), *dst, *tag,
190                     get_comm(*comm));
191 }
192
193 void mpi_recv_init__(void *buf, int* count, int* datatype, int* src, int* tag,
194                      int* comm, int* request, int* ierr) {
195   MPI_Request req;
196
197   *ierr = MPI_Recv_init(buf, *count, get_datatype(*datatype), *src, *tag,
198                         get_comm(*comm), &req);
199   if(*ierr == MPI_SUCCESS) {
200     *request = new_request(req);
201   }
202 }
203
204 void mpi_irecv__(void *buf, int* count, int* datatype, int* src, int* tag,
205                  int* comm, int* request, int* ierr) {
206   MPI_Request req;
207
208   *ierr = MPI_Irecv(buf, *count, get_datatype(*datatype), *src, *tag,
209                     get_comm(*comm), &req);
210   if(*ierr == MPI_SUCCESS) {
211     *request = new_request(req);
212   }
213 }
214
215 void mpi_recv__(void* buf, int* count, int* datatype, int* src,
216                 int* tag, int* comm, MPI_Status* status, int* ierr) {
217    *ierr = MPI_Recv(buf, *count, get_datatype(*datatype), *src, *tag,
218                     get_comm(*comm), status);
219 }
220
221 void mpi_start__(int* request, int* ierr) {
222   MPI_Request req = find_request(*request);
223
224   *ierr = MPI_Start(&req);
225 }
226
227 void mpi_startall__(int* count, int* requests, int* ierr) {
228   MPI_Request* reqs;
229   int i;
230
231   reqs = xbt_new(MPI_Request, *count);
232   for(i = 0; i < *count; i++) {
233     reqs[i] = find_request(requests[i]);
234   }
235   *ierr = MPI_Startall(*count, reqs);
236   free(reqs);
237 }
238
239 void mpi_wait__(int* request, MPI_Status* status, int* ierr) {
240    MPI_Request req = find_request(*request);
241    
242    *ierr = MPI_Wait(&req, status);
243 }
244
245 void mpi_waitany__(int* count, int* requests, int* index, MPI_Status* status, int* ierr) {
246   MPI_Request* reqs;
247   int i;
248
249   reqs = xbt_new(MPI_Request, *count);
250   for(i = 0; i < *count; i++) {
251     reqs[i] = find_request(requests[i]);
252   }
253   *ierr = MPI_Waitany(*count, reqs, index, status);
254   free(reqs);
255 }
256
257 void mpi_waitall__(int* count, int* requests, MPI_Status* status, int* ierr) {
258   MPI_Request* reqs;
259   int i;
260
261   reqs = xbt_new(MPI_Request, *count);
262   for(i = 0; i < *count; i++) {
263     reqs[i] = find_request(requests[i]);
264   }
265   *ierr = MPI_Waitall(*count, reqs, status);
266   free(reqs);
267 }
268
269 void mpi_barrier__(int* comm, int* ierr) {
270   *ierr = MPI_Barrier(get_comm(*comm));
271 }
272
273 void mpi_bcast__(void *buf, int* count, int* datatype, int* root, int* comm, int* ierr) {
274   *ierr = MPI_Bcast(buf, *count, get_datatype(*datatype), *root, get_comm(*comm));
275 }
276
277 void mpi_reduce__(void* sendbuf, void* recvbuf, int* count,
278                   int* datatype, int* op, int* root, int* comm, int* ierr) {
279   *ierr = MPI_Reduce(sendbuf, recvbuf, *count,
280                      get_datatype(*datatype), get_op(*op), *root, get_comm(*comm));
281 }
282
283 void mpi_allreduce__(void* sendbuf, void* recvbuf, int* count, int* datatype,
284                      int* op, int* comm, int* ierr) {
285   *ierr = MPI_Allreduce(sendbuf, recvbuf, *count, get_datatype(*datatype),
286                         get_op(*op), get_comm(*comm));
287 }
288
289 void mpi_scatter__(void* sendbuf, int* sendcount, int* sendtype,
290                    void* recvbuf, int* recvcount, int* recvtype, 
291                    int* root, int* comm, int* ierr) {
292   *ierr = MPI_Scatter(sendbuf, *sendcount, get_datatype(*sendtype),
293                       recvbuf, *recvcount, get_datatype(*recvtype), *root, get_comm(*comm));
294 }
295
296 void mpi_gather__(void* sendbuf, int* sendcount, int* sendtype,
297                   void* recvbuf, int* recvcount, int* recvtype,
298                   int* root, int* comm, int* ierr) {
299   *ierr = MPI_Gather(sendbuf, *sendcount, get_datatype(*sendtype),
300                      recvbuf, *recvcount, get_datatype(*recvtype), *root, get_comm(*comm));
301 }
302
303 void mpi_allgather__(void* sendbuf, int* sendcount, int* sendtype,
304                      void* recvbuf, int* recvcount, int* recvtype,
305                      int* comm, int* ierr) {
306   *ierr = MPI_Allgather(sendbuf, *sendcount, get_datatype(*sendtype),
307                         recvbuf, *recvcount, get_datatype(*recvtype), get_comm(*comm));
308 }
309
310 void mpi_scan__(void* sendbuf, void* recvbuf, int* count, int* datatype,
311                 int* op, int* comm, int* ierr) {
312   *ierr = MPI_Scan(sendbuf, recvbuf, *count, get_datatype(*datatype),
313                    get_op(*op), get_comm(*comm));
314 }
315
316 void mpi_alltoall__(void* sendbuf, int* sendcount, int* sendtype,
317                     void* recvbuf, int* recvcount, int* recvtype, int* comm, int* ierr) {
318   *ierr = MPI_Alltoall(sendbuf, *sendcount, get_datatype(*sendtype),
319                        recvbuf, *recvcount, get_datatype(*recvtype), get_comm(*comm));
320 }