1 /* Copyright (c) 2010-2019. The SimGrid Team. All rights reserved. */
3 /* This program is free software; you can redistribute it and/or modify it
4 * under the terms of the license (GNU LGPL) which comes with this package. */
7 #include "smpi_coll.hpp"
8 #include "smpi_comm.hpp"
9 #include "smpi_datatype.hpp"
10 #include "smpi_op.hpp"
11 #include "smpi_request.hpp"
13 extern "C" { // This should really use the C linkage to be usable from Fortran
15 void mpi_barrier_(int* comm, int* ierr) {
16 *ierr = MPI_Barrier(simgrid::smpi::Comm::f2c(*comm));
19 void mpi_bcast_(void *buf, int* count, int* datatype, int* root, int* comm, int* ierr) {
20 *ierr = MPI_Bcast(buf, *count, simgrid::smpi::Datatype::f2c(*datatype), *root, simgrid::smpi::Comm::f2c(*comm));
23 void mpi_reduce_(void* sendbuf, void* recvbuf, int* count, int* datatype, int* op, int* root, int* comm, int* ierr) {
24 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
25 sendbuf = static_cast<char *>( FORT_BOTTOM(sendbuf));
26 recvbuf = static_cast<char *>( FORT_BOTTOM(recvbuf));
27 *ierr = MPI_Reduce(sendbuf, recvbuf, *count, simgrid::smpi::Datatype::f2c(*datatype), simgrid::smpi::Op::f2c(*op), *root, simgrid::smpi::Comm::f2c(*comm));
30 void mpi_allreduce_(void* sendbuf, void* recvbuf, int* count, int* datatype, int* op, int* comm, int* ierr) {
31 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
32 *ierr = MPI_Allreduce(sendbuf, recvbuf, *count, simgrid::smpi::Datatype::f2c(*datatype), simgrid::smpi::Op::f2c(*op), simgrid::smpi::Comm::f2c(*comm));
35 void mpi_reduce_scatter_(void* sendbuf, void* recvbuf, int* recvcounts, int* datatype, int* op, int* comm, int* ierr) {
36 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
37 *ierr = MPI_Reduce_scatter(sendbuf, recvbuf, recvcounts, simgrid::smpi::Datatype::f2c(*datatype),
38 simgrid::smpi::Op::f2c(*op), simgrid::smpi::Comm::f2c(*comm));
41 void mpi_scatter_(void* sendbuf, int* sendcount, int* sendtype, void* recvbuf, int* recvcount, int* recvtype,
42 int* root, int* comm, int* ierr) {
43 recvbuf = static_cast<char *>( FORT_IN_PLACE(recvbuf));
44 *ierr = MPI_Scatter(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
45 recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*recvtype), *root, simgrid::smpi::Comm::f2c(*comm));
48 void mpi_scatterv_(void* sendbuf, int* sendcounts, int* displs, int* sendtype,
49 void* recvbuf, int* recvcount, int* recvtype, int* root, int* comm, int* ierr) {
50 recvbuf = static_cast<char *>( FORT_IN_PLACE(recvbuf));
51 *ierr = MPI_Scatterv(sendbuf, sendcounts, displs, simgrid::smpi::Datatype::f2c(*sendtype),
52 recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*recvtype), *root, simgrid::smpi::Comm::f2c(*comm));
55 void mpi_gather_(void* sendbuf, int* sendcount, int* sendtype, void* recvbuf, int* recvcount, int* recvtype,
56 int* root, int* comm, int* ierr) {
57 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
58 sendbuf = sendbuf!=MPI_IN_PLACE ? static_cast<char *>( FORT_BOTTOM(sendbuf)) : MPI_IN_PLACE;
59 recvbuf = static_cast<char *>( FORT_BOTTOM(recvbuf));
60 *ierr = MPI_Gather(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
61 recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*recvtype), *root, simgrid::smpi::Comm::f2c(*comm));
64 void mpi_gatherv_(void* sendbuf, int* sendcount, int* sendtype,
65 void* recvbuf, int* recvcounts, int* displs, int* recvtype, int* root, int* comm, int* ierr) {
66 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
67 sendbuf = sendbuf!=MPI_IN_PLACE ? static_cast<char *>( FORT_BOTTOM(sendbuf)) : MPI_IN_PLACE;
68 recvbuf = static_cast<char *>( FORT_BOTTOM(recvbuf));
69 *ierr = MPI_Gatherv(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
70 recvbuf, recvcounts, displs, simgrid::smpi::Datatype::f2c(*recvtype), *root, simgrid::smpi::Comm::f2c(*comm));
73 void mpi_allgather_(void* sendbuf, int* sendcount, int* sendtype, void* recvbuf, int* recvcount, int* recvtype,
74 int* comm, int* ierr) {
75 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
76 *ierr = MPI_Allgather(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
77 recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*recvtype), simgrid::smpi::Comm::f2c(*comm));
80 void mpi_allgatherv_(void* sendbuf, int* sendcount, int* sendtype,
81 void* recvbuf, int* recvcounts,int* displs, int* recvtype, int* comm, int* ierr) {
82 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
83 *ierr = MPI_Allgatherv(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
84 recvbuf, recvcounts, displs, simgrid::smpi::Datatype::f2c(*recvtype), simgrid::smpi::Comm::f2c(*comm));
87 void mpi_scan_(void* sendbuf, void* recvbuf, int* count, int* datatype, int* op, int* comm, int* ierr) {
88 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
89 *ierr = MPI_Scan(sendbuf, recvbuf, *count, simgrid::smpi::Datatype::f2c(*datatype),
90 simgrid::smpi::Op::f2c(*op), simgrid::smpi::Comm::f2c(*comm));
93 void mpi_alltoall_(void* sendbuf, int* sendcount, int* sendtype,
94 void* recvbuf, int* recvcount, int* recvtype, int* comm, int* ierr) {
95 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
96 *ierr = MPI_Alltoall(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
97 recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*recvtype), simgrid::smpi::Comm::f2c(*comm));
100 void mpi_alltoallv_(void* sendbuf, int* sendcounts, int* senddisps, int* sendtype,
101 void* recvbuf, int* recvcounts, int* recvdisps, int* recvtype, int* comm, int* ierr) {
102 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
103 *ierr = MPI_Alltoallv(sendbuf, sendcounts, senddisps, simgrid::smpi::Datatype::f2c(*sendtype),
104 recvbuf, recvcounts, recvdisps, simgrid::smpi::Datatype::f2c(*recvtype), simgrid::smpi::Comm::f2c(*comm));
107 void mpi_reduce_local_ (void *inbuf, void *inoutbuf, int* count, int* datatype, int* op, int* ierr){
109 *ierr = MPI_Reduce_local(inbuf, inoutbuf, *count, simgrid::smpi::Datatype::f2c(*datatype), simgrid::smpi::Op::f2c(*op));
112 void mpi_reduce_scatter_block_ (void *sendbuf, void *recvbuf, int* recvcount, int* datatype, int* op, int* comm,
115 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
116 *ierr = MPI_Reduce_scatter_block(sendbuf, recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*datatype), simgrid::smpi::Op::f2c(*op),
117 simgrid::smpi::Comm::f2c(*comm));
120 void mpi_alltoallw_ ( void *sendbuf, int *sendcnts, int *sdispls, int* old_sendtypes, void *recvbuf, int *recvcnts,
121 int *rdispls, int* old_recvtypes, int* comm, int* ierr){
122 int size = simgrid::smpi::Comm::f2c(*comm)->size();
123 MPI_Datatype* sendtypes = new MPI_Datatype[size];
124 MPI_Datatype* recvtypes = new MPI_Datatype[size];
125 for(int i=0; i< size; i++){
126 if(FORT_IN_PLACE(sendbuf)!=MPI_IN_PLACE)
127 sendtypes[i] = simgrid::smpi::Datatype::f2c(old_sendtypes[i]);
128 recvtypes[i] = simgrid::smpi::Datatype::f2c(old_recvtypes[i]);
130 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
131 *ierr = MPI_Alltoallw( sendbuf, sendcnts, sdispls, sendtypes, recvbuf, recvcnts, rdispls,
132 recvtypes, simgrid::smpi::Comm::f2c(*comm));
137 void mpi_exscan_ (void *sendbuf, void *recvbuf, int* count, int* datatype, int* op, int* comm, int* ierr){
138 *ierr = MPI_Exscan(sendbuf, recvbuf, *count, simgrid::smpi::Datatype::f2c(*datatype), simgrid::smpi::Op::f2c(*op), simgrid::smpi::Comm::f2c(*comm));
141 void mpi_ibarrier_(int* comm, int* request, int* ierr) {
143 *ierr = MPI_Ibarrier(simgrid::smpi::Comm::f2c(*comm), &req);
144 if(*ierr == MPI_SUCCESS) {
145 *request = req->add_f();
149 void mpi_ibcast_(void *buf, int* count, int* datatype, int* root, int* comm, int* request, int* ierr) {
151 *ierr = MPI_Ibcast(buf, *count, simgrid::smpi::Datatype::f2c(*datatype), *root, simgrid::smpi::Comm::f2c(*comm), &req);
152 if(*ierr == MPI_SUCCESS) {
153 *request = req->add_f();
157 void mpi_ireduce_(void* sendbuf, void* recvbuf, int* count, int* datatype, int* op, int* root, int* comm, int* request, int* ierr) {
159 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
160 sendbuf = static_cast<char *>( FORT_BOTTOM(sendbuf));
161 recvbuf = static_cast<char *>( FORT_BOTTOM(recvbuf));
162 *ierr = MPI_Ireduce(sendbuf, recvbuf, *count, simgrid::smpi::Datatype::f2c(*datatype), simgrid::smpi::Op::f2c(*op), *root, simgrid::smpi::Comm::f2c(*comm), &req);
163 if(*ierr == MPI_SUCCESS) {
164 *request = req->add_f();
168 void mpi_iallreduce_(void* sendbuf, void* recvbuf, int* count, int* datatype, int* op, int* comm, int* request, int* ierr) {
170 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
171 *ierr = MPI_Iallreduce(sendbuf, recvbuf, *count, simgrid::smpi::Datatype::f2c(*datatype), simgrid::smpi::Op::f2c(*op), simgrid::smpi::Comm::f2c(*comm), &req);
172 if(*ierr == MPI_SUCCESS) {
173 *request = req->add_f();
177 void mpi_ireduce_scatter_(void* sendbuf, void* recvbuf, int* recvcounts, int* datatype, int* op, int* comm, int* request, int* ierr) {
179 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
180 *ierr = MPI_Ireduce_scatter(sendbuf, recvbuf, recvcounts, simgrid::smpi::Datatype::f2c(*datatype),
181 simgrid::smpi::Op::f2c(*op), simgrid::smpi::Comm::f2c(*comm), &req);
182 if(*ierr == MPI_SUCCESS) {
183 *request = req->add_f();
187 void mpi_iscatter_(void* sendbuf, int* sendcount, int* sendtype, void* recvbuf, int* recvcount, int* recvtype,
188 int* root, int* comm, int* request, int* ierr) {
190 recvbuf = static_cast<char *>( FORT_IN_PLACE(recvbuf));
191 *ierr = MPI_Iscatter(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
192 recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*recvtype), *root, simgrid::smpi::Comm::f2c(*comm), &req);
193 if(*ierr == MPI_SUCCESS) {
194 *request = req->add_f();
198 void mpi_iscatterv_(void* sendbuf, int* sendcounts, int* displs, int* sendtype,
199 void* recvbuf, int* recvcount, int* recvtype, int* root, int* comm, int* request, int* ierr) {
201 recvbuf = static_cast<char *>( FORT_IN_PLACE(recvbuf));
202 *ierr = MPI_Iscatterv(sendbuf, sendcounts, displs, simgrid::smpi::Datatype::f2c(*sendtype),
203 recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*recvtype), *root, simgrid::smpi::Comm::f2c(*comm), &req);
204 if(*ierr == MPI_SUCCESS) {
205 *request = req->add_f();
209 void mpi_igather_(void* sendbuf, int* sendcount, int* sendtype, void* recvbuf, int* recvcount, int* recvtype,
210 int* root, int* comm, int* request, int* ierr) {
212 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
213 sendbuf = sendbuf!=MPI_IN_PLACE ? static_cast<char *>( FORT_BOTTOM(sendbuf)) : MPI_IN_PLACE;
214 recvbuf = static_cast<char *>( FORT_BOTTOM(recvbuf));
215 *ierr = MPI_Igather(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
216 recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*recvtype), *root, simgrid::smpi::Comm::f2c(*comm), &req);
217 if(*ierr == MPI_SUCCESS) {
218 *request = req->add_f();
222 void mpi_igatherv_(void* sendbuf, int* sendcount, int* sendtype,
223 void* recvbuf, int* recvcounts, int* displs, int* recvtype, int* root, int* comm, int* request, int* ierr) {
225 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
226 sendbuf = sendbuf!=MPI_IN_PLACE ? static_cast<char *>( FORT_BOTTOM(sendbuf)) : MPI_IN_PLACE;
227 recvbuf = static_cast<char *>( FORT_BOTTOM(recvbuf));
228 *ierr = MPI_Igatherv(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
229 recvbuf, recvcounts, displs, simgrid::smpi::Datatype::f2c(*recvtype), *root, simgrid::smpi::Comm::f2c(*comm), &req);
230 if(*ierr == MPI_SUCCESS) {
231 *request = req->add_f();
235 void mpi_iallgather_(void* sendbuf, int* sendcount, int* sendtype, void* recvbuf, int* recvcount, int* recvtype,
236 int* comm, int* request, int* ierr) {
238 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
239 *ierr = MPI_Iallgather(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
240 recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*recvtype), simgrid::smpi::Comm::f2c(*comm), &req);
241 if(*ierr == MPI_SUCCESS) {
242 *request = req->add_f();
246 void mpi_iallgatherv_(void* sendbuf, int* sendcount, int* sendtype,
247 void* recvbuf, int* recvcounts,int* displs, int* recvtype, int* comm, int* request, int* ierr) {
249 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
250 *ierr = MPI_Iallgatherv(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
251 recvbuf, recvcounts, displs, simgrid::smpi::Datatype::f2c(*recvtype), simgrid::smpi::Comm::f2c(*comm), &req);
252 if(*ierr == MPI_SUCCESS) {
253 *request = req->add_f();
257 void mpi_iscan_(void* sendbuf, void* recvbuf, int* count, int* datatype, int* op, int* comm, int* request, int* ierr) {
259 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
260 *ierr = MPI_Iscan(sendbuf, recvbuf, *count, simgrid::smpi::Datatype::f2c(*datatype),
261 simgrid::smpi::Op::f2c(*op), simgrid::smpi::Comm::f2c(*comm), &req);
262 if(*ierr == MPI_SUCCESS) {
263 *request = req->add_f();
267 void mpi_ialltoall_(void* sendbuf, int* sendcount, int* sendtype,
268 void* recvbuf, int* recvcount, int* recvtype, int* comm, int* request, int* ierr) {
270 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
271 *ierr = MPI_Ialltoall(sendbuf, *sendcount, simgrid::smpi::Datatype::f2c(*sendtype),
272 recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*recvtype), simgrid::smpi::Comm::f2c(*comm), &req);
273 if(*ierr == MPI_SUCCESS) {
274 *request = req->add_f();
278 void mpi_ialltoallv_(void* sendbuf, int* sendcounts, int* senddisps, int* sendtype,
279 void* recvbuf, int* recvcounts, int* recvdisps, int* recvtype, int* comm, int* request, int* ierr) {
281 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
282 *ierr = MPI_Ialltoallv(sendbuf, sendcounts, senddisps, simgrid::smpi::Datatype::f2c(*sendtype),
283 recvbuf, recvcounts, recvdisps, simgrid::smpi::Datatype::f2c(*recvtype), simgrid::smpi::Comm::f2c(*comm), &req);
284 if(*ierr == MPI_SUCCESS) {
285 *request = req->add_f();
289 void mpi_ireduce_scatter_block_ (void *sendbuf, void *recvbuf, int* recvcount, int* datatype, int* op, int* comm,
290 int* request, int* ierr)
293 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
294 *ierr = MPI_Ireduce_scatter_block(sendbuf, recvbuf, *recvcount, simgrid::smpi::Datatype::f2c(*datatype), simgrid::smpi::Op::f2c(*op),
295 simgrid::smpi::Comm::f2c(*comm), &req);
296 if(*ierr == MPI_SUCCESS) {
297 *request = req->add_f();
301 void mpi_ialltoallw_ ( void *sendbuf, int *sendcnts, int *sdispls, int* old_sendtypes, void *recvbuf, int *recvcnts,
302 int *rdispls, int* old_recvtypes, int* comm, int* request, int* ierr){
304 int size = simgrid::smpi::Comm::f2c(*comm)->size();
305 MPI_Datatype* sendtypes = new MPI_Datatype[size];
306 MPI_Datatype* recvtypes = new MPI_Datatype[size];
307 for(int i=0; i< size; i++){
308 if(FORT_IN_PLACE(sendbuf)!=MPI_IN_PLACE)
309 sendtypes[i] = simgrid::smpi::Datatype::f2c(old_sendtypes[i]);
310 recvtypes[i] = simgrid::smpi::Datatype::f2c(old_recvtypes[i]);
312 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
313 *ierr = MPI_Ialltoallw( sendbuf, sendcnts, sdispls, sendtypes, recvbuf, recvcnts, rdispls,
314 recvtypes, simgrid::smpi::Comm::f2c(*comm), &req);
315 if(*ierr == MPI_SUCCESS) {
316 *request = req->add_f();
322 void mpi_iexscan_ (void *sendbuf, void *recvbuf, int* count, int* datatype, int* op, int* comm, int* request, int* ierr){
324 sendbuf = static_cast<char *>( FORT_IN_PLACE(sendbuf));
325 *ierr = MPI_Iexscan(sendbuf, recvbuf, *count, simgrid::smpi::Datatype::f2c(*datatype), simgrid::smpi::Op::f2c(*op), simgrid::smpi::Comm::f2c(*comm), &req);
326 if(*ierr == MPI_SUCCESS) {
327 *request = req->add_f();