Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Convert enum smpi_process_state to enum class.
[simgrid.git] / src / smpi / bindings / smpi_pmpi_comm.cpp
1 /* Copyright (c) 2007-2018. The SimGrid Team. All rights reserved.          */
2
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. */
5
6 #include <climits>
7
8 #include "private.hpp"
9 #include "smpi_comm.hpp"
10 #include "smpi_process.hpp"
11
12 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(smpi_pmpi);
13
14 /* PMPI User level calls */
15
16 int PMPI_Comm_rank(MPI_Comm comm, int *rank)
17 {
18   if (comm == MPI_COMM_NULL) {
19     return MPI_ERR_COMM;
20   } else if (rank == nullptr) {
21     return MPI_ERR_ARG;
22   } else {
23     *rank = comm->rank();
24     return MPI_SUCCESS;
25   }
26 }
27
28 int PMPI_Comm_size(MPI_Comm comm, int *size)
29 {
30   if (comm == MPI_COMM_NULL) {
31     return MPI_ERR_COMM;
32   } else if (size == nullptr) {
33     return MPI_ERR_ARG;
34   } else {
35     *size = comm->size();
36     return MPI_SUCCESS;
37   }
38 }
39
40 int PMPI_Comm_get_name (MPI_Comm comm, char* name, int* len)
41 {
42   if (comm == MPI_COMM_NULL)  {
43     return MPI_ERR_COMM;
44   } else if (name == nullptr || len == nullptr)  {
45     return MPI_ERR_ARG;
46   } else {
47     comm->get_name(name, len);
48     return MPI_SUCCESS;
49   }
50 }
51
52 int PMPI_Comm_group(MPI_Comm comm, MPI_Group * group)
53 {
54   if (comm == MPI_COMM_NULL) {
55     return MPI_ERR_COMM;
56   } else if (group == nullptr) {
57     return MPI_ERR_ARG;
58   } else {
59     *group = comm->group();
60     if (*group != MPI_COMM_WORLD->group() && *group != MPI_GROUP_NULL && *group != MPI_GROUP_EMPTY)
61       (*group)->ref();
62     return MPI_SUCCESS;
63   }
64 }
65
66 int PMPI_Comm_compare(MPI_Comm comm1, MPI_Comm comm2, int *result)
67 {
68   if (comm1 == MPI_COMM_NULL || comm2 == MPI_COMM_NULL) {
69     return MPI_ERR_COMM;
70   } else if (result == nullptr) {
71     return MPI_ERR_ARG;
72   } else {
73     if (comm1 == comm2) {       /* Same communicators means same groups */
74       *result = MPI_IDENT;
75     } else {
76       *result = comm1->group()->compare(comm2->group());
77       if (*result == MPI_IDENT) {
78         *result = MPI_CONGRUENT;
79       }
80     }
81     return MPI_SUCCESS;
82   }
83 }
84
85 int PMPI_Comm_dup(MPI_Comm comm, MPI_Comm * newcomm)
86 {
87   if (comm == MPI_COMM_NULL) {
88     return MPI_ERR_COMM;
89   } else if (newcomm == nullptr) {
90     return MPI_ERR_ARG;
91   } else {
92     return comm->dup(newcomm);
93   }
94 }
95
96 int PMPI_Comm_create(MPI_Comm comm, MPI_Group group, MPI_Comm * newcomm)
97 {
98   if (comm == MPI_COMM_NULL) {
99     return MPI_ERR_COMM;
100   } else if (group == MPI_GROUP_NULL) {
101     return MPI_ERR_GROUP;
102   } else if (newcomm == nullptr) {
103     return MPI_ERR_ARG;
104   } else if (group->rank(simgrid::s4u::this_actor::get_pid()) == MPI_UNDEFINED) {
105     *newcomm= MPI_COMM_NULL;
106     return MPI_SUCCESS;
107   }else{
108     group->ref();
109     *newcomm = new simgrid::smpi::Comm(group, nullptr);
110     return MPI_SUCCESS;
111   }
112 }
113
114 int PMPI_Comm_free(MPI_Comm * comm)
115 {
116   if (comm == nullptr) {
117     return MPI_ERR_ARG;
118   } else if (*comm == MPI_COMM_NULL) {
119     return MPI_ERR_COMM;
120   } else {
121     simgrid::smpi::Comm::destroy(*comm);
122     *comm = MPI_COMM_NULL;
123     return MPI_SUCCESS;
124   }
125 }
126
127 int PMPI_Comm_disconnect(MPI_Comm * comm)
128 {
129   /* TODO: wait until all communication in comm are done */
130   if (comm == nullptr) {
131     return MPI_ERR_ARG;
132   } else if (*comm == MPI_COMM_NULL) {
133     return MPI_ERR_COMM;
134   } else {
135     simgrid::smpi::Comm::destroy(*comm);
136     *comm = MPI_COMM_NULL;
137     return MPI_SUCCESS;
138   }
139 }
140
141 int PMPI_Comm_split(MPI_Comm comm, int color, int key, MPI_Comm* comm_out)
142 {
143   int retval = 0;
144   smpi_bench_end();
145
146   if (comm_out == nullptr) {
147     retval = MPI_ERR_ARG;
148   } else if (comm == MPI_COMM_NULL) {
149     retval = MPI_ERR_COMM;
150   } else {
151     *comm_out = comm->split(color, key);
152     retval = MPI_SUCCESS;
153   }
154   smpi_bench_begin();
155
156   return retval;
157 }
158
159 int PMPI_Comm_split_type(MPI_Comm comm, int split_type, int key, MPI_Info info, MPI_Comm *newcomm)
160 {
161   int retval = 0;
162   smpi_bench_end();
163
164   if (newcomm == nullptr) {
165     retval = MPI_ERR_ARG;
166   } else if (comm == MPI_COMM_NULL) {
167     retval = MPI_ERR_COMM;
168   } else {
169     *newcomm = comm->split_type(split_type, key, info);
170     retval = MPI_SUCCESS;
171   }
172   smpi_bench_begin();
173
174   return retval;
175 }
176
177 int PMPI_Comm_create_group(MPI_Comm comm, MPI_Group group, int, MPI_Comm* comm_out)
178 {
179   int retval = 0;
180   smpi_bench_end();
181
182   if (comm_out == nullptr) {
183     retval = MPI_ERR_ARG;
184   } else if (comm == MPI_COMM_NULL) {
185     retval = MPI_ERR_COMM;
186   } else {
187     retval = MPI_Comm_create(comm, group, comm_out);
188   }
189   smpi_bench_begin();
190
191   return retval;
192 }
193
194 MPI_Comm PMPI_Comm_f2c(MPI_Fint comm){
195   return static_cast<MPI_Comm>(simgrid::smpi::Comm::f2c(comm));
196 }
197
198 MPI_Fint PMPI_Comm_c2f(MPI_Comm comm){
199   return comm->c2f();
200 }
201
202 int PMPI_Comm_get_attr (MPI_Comm comm, int comm_keyval, void *attribute_val, int *flag)
203 {
204   return PMPI_Attr_get(comm, comm_keyval, attribute_val,flag);
205 }
206
207 int PMPI_Comm_set_attr (MPI_Comm comm, int comm_keyval, void *attribute_val)
208 {
209   return PMPI_Attr_put(comm, comm_keyval, attribute_val);
210 }
211
212 int PMPI_Comm_delete_attr (MPI_Comm comm, int comm_keyval)
213 {
214   return PMPI_Attr_delete(comm, comm_keyval);
215 }
216
217 int PMPI_Comm_create_keyval(MPI_Comm_copy_attr_function* copy_fn, MPI_Comm_delete_attr_function* delete_fn, int* keyval,
218                             void* extra_state)
219 {
220   return PMPI_Keyval_create(copy_fn, delete_fn, keyval, extra_state);
221 }
222
223 int PMPI_Comm_free_keyval(int* keyval) {
224   return PMPI_Keyval_free(keyval);
225 }
226
227 int PMPI_Attr_delete(MPI_Comm comm, int keyval) {
228   if(keyval == MPI_TAG_UB||keyval == MPI_HOST||keyval == MPI_IO ||keyval == MPI_WTIME_IS_GLOBAL||keyval == MPI_APPNUM
229        ||keyval == MPI_UNIVERSE_SIZE||keyval == MPI_LASTUSEDCODE)
230     return MPI_ERR_ARG;
231   else if (comm==MPI_COMM_NULL)
232     return MPI_ERR_COMM;
233   else
234     return comm->attr_delete<simgrid::smpi::Comm>(keyval);
235 }
236
237 int PMPI_Attr_get(MPI_Comm comm, int keyval, void* attr_value, int* flag) {
238   static int one = 1;
239   static int zero = 0;
240   static int tag_ub = INT_MAX;
241   static int last_used_code = MPI_ERR_LASTCODE;
242
243   if (comm==MPI_COMM_NULL){
244     *flag = 0;
245     return MPI_ERR_COMM;
246   }
247
248   switch (keyval) {
249   case MPI_HOST:
250   case MPI_IO:
251   case MPI_APPNUM:
252     *flag = 1;
253     *static_cast<int**>(attr_value) = &zero;
254     return MPI_SUCCESS;
255   case MPI_UNIVERSE_SIZE:
256     *flag = 1;
257     *static_cast<int**>(attr_value) = &smpi_universe_size;
258     return MPI_SUCCESS;
259   case MPI_LASTUSEDCODE:
260     *flag = 1;
261     *static_cast<int**>(attr_value) = &last_used_code;
262     return MPI_SUCCESS;
263   case MPI_TAG_UB:
264     *flag=1;
265     *static_cast<int**>(attr_value) = &tag_ub;
266     return MPI_SUCCESS;
267   case MPI_WTIME_IS_GLOBAL:
268     *flag = 1;
269     *static_cast<int**>(attr_value) = &one;
270     return MPI_SUCCESS;
271   default:
272     return comm->attr_get<simgrid::smpi::Comm>(keyval, attr_value, flag);
273   }
274 }
275
276 int PMPI_Attr_put(MPI_Comm comm, int keyval, void* attr_value) {
277   if(keyval == MPI_TAG_UB||keyval == MPI_HOST||keyval == MPI_IO ||keyval == MPI_WTIME_IS_GLOBAL||keyval == MPI_APPNUM
278        ||keyval == MPI_UNIVERSE_SIZE||keyval == MPI_LASTUSEDCODE)
279     return MPI_ERR_ARG;
280   else if (comm==MPI_COMM_NULL)
281     return MPI_ERR_COMM;
282   else
283   return comm->attr_put<simgrid::smpi::Comm>(keyval, attr_value);
284 }