1 /* Copyright (c) 2009-2023. 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. */
8 #include "smpi_datatype.hpp"
9 #include "src/smpi/include/smpi_actor.hpp"
11 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(smpi_op, smpi, "Logging specific to SMPI (op)");
13 #define MAX_OP(a, b) (b) = (a) < (b) ? (b) : (a)
14 #define MIN_OP(a, b) (b) = (a) < (b) ? (a) : (b)
15 #define SUM_OP(a, b) (b) += (a)
16 #define SUM_OP_COMPLEX(a, b) \
18 ((b).value) += ((a).value); \
19 ((b).index) += ((a).index); \
21 #define PROD_OP(a, b) (b) *= (a)
22 #define PROD_OP_COMPLEX(a, b) \
24 ((b).value) *= ((a).value); \
25 ((b).index) *= ((a).index); \
27 #define LAND_OP(a, b) (b) = static_cast<std::remove_reference_t<decltype(b)>>((a) && (b))
28 #define LOR_OP(a, b) (b) = static_cast<std::remove_reference_t<decltype(b)>>((a) || (b))
29 #define LXOR_OP(a, b) (b) = static_cast<std::remove_reference_t<decltype(b)>>(bool(a) != bool(b))
30 #define BAND_OP(a, b) (b) &= (a)
31 #define BOR_OP(a, b) (b) |= (a)
32 #define BXOR_OP(a, b) (b) ^= (a)
33 #define MAXLOC_OP(a, b) \
34 (b) = ((a).value) < ((b).value) ? (b) : (((a).value) == ((b).value) ? (((a).index) < ((b).index) ? (a) : (b)) : (a))
35 #define MINLOC_OP(a, b) \
36 (b) = ((a).value) < ((b).value) ? (a) : (((a).value) == ((b).value) ? (((a).index) < ((b).index) ? (a) : (b)) : (b))
38 #define APPLY_FUNC(a, b, length, type, func) \
41 type* x = (type*)(a); \
42 type* y = (type*)(b); \
43 for(i = 0; i < *(length); i++) { \
48 #define APPLY_BEGIN_OP_LOOP() \
49 MPI_Datatype datatype_base = *datatype; \
50 while (datatype_base->duplicated_datatype() != MPI_DATATYPE_NULL) \
51 datatype_base = datatype_base->duplicated_datatype();
53 #define APPLY_OP_LOOP(dtype, type, op) \
54 if (datatype_base == (dtype)) { \
55 APPLY_FUNC(a, b, length, type, op) \
58 #define APPLY_BASIC_OP_LOOP(op)\
59 APPLY_OP_LOOP(MPI_CHAR, char,op)\
60 APPLY_OP_LOOP(MPI_SHORT, short,op)\
61 APPLY_OP_LOOP(MPI_INT, int,op)\
62 APPLY_OP_LOOP(MPI_LONG, long,op)\
63 APPLY_OP_LOOP(MPI_LONG_LONG, long long,op)\
64 APPLY_OP_LOOP(MPI_SIGNED_CHAR, signed char,op)\
65 APPLY_OP_LOOP(MPI_UNSIGNED_CHAR, unsigned char,op)\
66 APPLY_OP_LOOP(MPI_UNSIGNED_SHORT, unsigned short,op)\
67 APPLY_OP_LOOP(MPI_UNSIGNED, unsigned int,op)\
68 APPLY_OP_LOOP(MPI_UNSIGNED_LONG, unsigned long,op)\
69 APPLY_OP_LOOP(MPI_UNSIGNED_LONG_LONG, unsigned long long,op)\
70 APPLY_OP_LOOP(MPI_WCHAR, wchar_t,op)\
71 APPLY_OP_LOOP(MPI_INT8_T, int8_t,op)\
72 APPLY_OP_LOOP(MPI_INT16_T, int16_t,op)\
73 APPLY_OP_LOOP(MPI_INT32_T, int32_t,op)\
74 APPLY_OP_LOOP(MPI_INT64_T, int64_t,op)\
75 APPLY_OP_LOOP(MPI_UINT8_T, uint8_t,op)\
76 APPLY_OP_LOOP(MPI_UINT16_T, uint16_t,op)\
77 APPLY_OP_LOOP(MPI_UINT32_T, uint32_t,op)\
78 APPLY_OP_LOOP(MPI_UINT64_T, uint64_t,op)\
79 APPLY_OP_LOOP(MPI_AINT, MPI_Aint,op)\
80 APPLY_OP_LOOP(MPI_OFFSET, MPI_Offset,op)\
81 APPLY_OP_LOOP(MPI_INTEGER1, int,op)\
82 APPLY_OP_LOOP(MPI_INTEGER2, int16_t,op)\
83 APPLY_OP_LOOP(MPI_INTEGER4, int32_t,op)\
84 APPLY_OP_LOOP(MPI_INTEGER8, int64_t,op)\
85 APPLY_OP_LOOP(MPI_COUNT, long long,op)
88 #define APPLY_BOOL_OP_LOOP(op)\
89 APPLY_OP_LOOP(MPI_C_BOOL, bool,op)
91 #define APPLY_BYTE_OP_LOOP(op)\
92 APPLY_OP_LOOP(MPI_BYTE, int8_t,op)
94 #define APPLY_FLOAT_OP_LOOP(op)\
95 APPLY_OP_LOOP(MPI_FLOAT, float,op)\
96 APPLY_OP_LOOP(MPI_DOUBLE, double,op)\
97 APPLY_OP_LOOP(MPI_LONG_DOUBLE, long double,op)\
98 APPLY_OP_LOOP(MPI_REAL, float,op)\
99 APPLY_OP_LOOP(MPI_REAL4, float,op)\
100 APPLY_OP_LOOP(MPI_REAL8, double,op)\
101 APPLY_OP_LOOP(MPI_REAL16, long double,op)
103 #define APPLY_COMPLEX_OP_LOOP(op)\
104 APPLY_OP_LOOP(MPI_C_FLOAT_COMPLEX, float _Complex,op)\
105 APPLY_OP_LOOP(MPI_C_DOUBLE_COMPLEX, double _Complex,op)\
106 APPLY_OP_LOOP(MPI_C_LONG_DOUBLE_COMPLEX, long double _Complex,op)
108 #define APPLY_PAIR_OP_LOOP(op)\
109 APPLY_OP_LOOP(MPI_FLOAT_INT, float_int,op)\
110 APPLY_OP_LOOP(MPI_LONG_INT, long_int,op)\
111 APPLY_OP_LOOP(MPI_DOUBLE_INT, double_int,op)\
112 APPLY_OP_LOOP(MPI_SHORT_INT, short_int,op)\
113 APPLY_OP_LOOP(MPI_2INT, int_int,op)\
114 APPLY_OP_LOOP(MPI_2FLOAT, float_float,op)\
115 APPLY_OP_LOOP(MPI_2DOUBLE, double_double,op)\
116 APPLY_OP_LOOP(MPI_LONG_DOUBLE_INT, long_double_int,op)\
117 APPLY_OP_LOOP(MPI_2LONG, long_long,op)\
118 APPLY_OP_LOOP(MPI_COMPLEX8, float_float,op)\
119 APPLY_OP_LOOP(MPI_COMPLEX16, double_double,op)\
120 APPLY_OP_LOOP(MPI_COMPLEX32, double_double,op)
122 #define APPLY_END_OP_LOOP(op) \
124 xbt_die("Failed to apply " _XBT_STRINGIFY(op) " to type %s", (*datatype)->name().c_str()); \
127 static void max_func(void *a, void *b, int *length, MPI_Datatype * datatype)
129 APPLY_BEGIN_OP_LOOP()
130 APPLY_BASIC_OP_LOOP(MAX_OP)
131 APPLY_FLOAT_OP_LOOP(MAX_OP)
132 APPLY_END_OP_LOOP(MAX_OP)
135 static void min_func(void *a, void *b, int *length, MPI_Datatype * datatype)
137 APPLY_BEGIN_OP_LOOP()
138 APPLY_BASIC_OP_LOOP(MIN_OP)
139 APPLY_FLOAT_OP_LOOP(MIN_OP)
140 APPLY_END_OP_LOOP(MIN_OP)
143 static void sum_func(void *a, void *b, int *length, MPI_Datatype * datatype)
145 APPLY_BEGIN_OP_LOOP()
146 APPLY_BASIC_OP_LOOP(SUM_OP)
147 APPLY_FLOAT_OP_LOOP(SUM_OP)
148 APPLY_COMPLEX_OP_LOOP(SUM_OP)
149 APPLY_PAIR_OP_LOOP(SUM_OP_COMPLEX)
150 APPLY_END_OP_LOOP(SUM_OP)
153 static void prod_func(void *a, void *b, int *length, MPI_Datatype * datatype)
155 APPLY_BEGIN_OP_LOOP()
156 APPLY_BASIC_OP_LOOP(PROD_OP)
157 APPLY_FLOAT_OP_LOOP(PROD_OP)
158 APPLY_COMPLEX_OP_LOOP(PROD_OP)
159 APPLY_PAIR_OP_LOOP(PROD_OP_COMPLEX)
160 APPLY_END_OP_LOOP(PROD_OP)
163 static void land_func(void *a, void *b, int *length, MPI_Datatype * datatype)
165 APPLY_BEGIN_OP_LOOP()
166 APPLY_BASIC_OP_LOOP(LAND_OP)
167 APPLY_FLOAT_OP_LOOP(LAND_OP)
168 APPLY_BOOL_OP_LOOP(LAND_OP)
169 APPLY_END_OP_LOOP(LAND_OP)
172 static void lor_func(void *a, void *b, int *length, MPI_Datatype * datatype)
174 APPLY_BEGIN_OP_LOOP()
175 APPLY_BASIC_OP_LOOP(LOR_OP)
176 APPLY_FLOAT_OP_LOOP(LOR_OP)
177 APPLY_BOOL_OP_LOOP(LOR_OP)
178 APPLY_END_OP_LOOP(LOR_OP)
181 static void lxor_func(void *a, void *b, int *length, MPI_Datatype * datatype)
183 APPLY_BEGIN_OP_LOOP()
184 APPLY_BASIC_OP_LOOP(LXOR_OP)
185 APPLY_FLOAT_OP_LOOP(LXOR_OP)
186 APPLY_BOOL_OP_LOOP(LXOR_OP)
187 APPLY_END_OP_LOOP(LXOR_OP)
190 static void band_func(void *a, void *b, int *length, MPI_Datatype * datatype)
192 APPLY_BEGIN_OP_LOOP()
193 APPLY_BASIC_OP_LOOP(BAND_OP)
194 APPLY_BOOL_OP_LOOP(BAND_OP)
195 APPLY_BYTE_OP_LOOP(BAND_OP)
196 APPLY_END_OP_LOOP(BAND_OP)
199 static void bor_func(void *a, void *b, int *length, MPI_Datatype * datatype)
201 APPLY_BEGIN_OP_LOOP()
202 APPLY_BASIC_OP_LOOP(BOR_OP)
203 APPLY_BOOL_OP_LOOP(BOR_OP)
204 APPLY_BYTE_OP_LOOP(BOR_OP)
205 APPLY_END_OP_LOOP(BOR_OP)
208 static void bxor_func(void *a, void *b, int *length, MPI_Datatype * datatype)
210 APPLY_BEGIN_OP_LOOP()
211 APPLY_BASIC_OP_LOOP(BXOR_OP)
212 APPLY_BOOL_OP_LOOP(BXOR_OP)
213 APPLY_BYTE_OP_LOOP(BXOR_OP)
214 APPLY_END_OP_LOOP(BXOR_OP)
217 static void minloc_func(void *a, void *b, int *length, MPI_Datatype * datatype)
219 APPLY_BEGIN_OP_LOOP()
220 APPLY_PAIR_OP_LOOP(MINLOC_OP)
221 APPLY_END_OP_LOOP(MINLOC_OP)
224 static void maxloc_func(void *a, void *b, int *length, MPI_Datatype * datatype)
226 APPLY_BEGIN_OP_LOOP()
227 APPLY_PAIR_OP_LOOP(MAXLOC_OP)
228 APPLY_END_OP_LOOP(MAXLOC_OP)
231 static void replace_func(void *a, void *b, int *length, MPI_Datatype * datatype)
233 memcpy(b, a, *length * (*datatype)->size());
236 static void no_func(void*, void*, int*, MPI_Datatype*)
238 /* obviously a no-op */
241 #define CREATE_MPI_OP(name, func, types) \
242 SMPI_Op _XBT_CONCAT(smpi_MPI_, name)(&(func), true, true, types, _XBT_STRINGIFY(MPI_##name));
244 #define MAX_TYPES DT_FLAG_C_INTEGER|DT_FLAG_F_INTEGER|DT_FLAG_FP|DT_FLAG_MULTILANG
245 #define LAND_TYPES DT_FLAG_C_INTEGER|DT_FLAG_FP|DT_FLAG_LOGICAL|DT_FLAG_MULTILANG
246 #define BAND_TYPES DT_FLAG_C_INTEGER|DT_FLAG_F_INTEGER|DT_FLAG_BYTE|DT_FLAG_MULTILANG
248 CREATE_MPI_OP(MAX, max_func, MAX_TYPES)
249 CREATE_MPI_OP(MIN, min_func, MAX_TYPES)
250 CREATE_MPI_OP(SUM, sum_func, MAX_TYPES|DT_FLAG_COMPLEX)
251 CREATE_MPI_OP(PROD, prod_func, MAX_TYPES|DT_FLAG_COMPLEX)
252 CREATE_MPI_OP(LAND, land_func, LAND_TYPES)
253 CREATE_MPI_OP(LOR, lor_func, LAND_TYPES)
254 CREATE_MPI_OP(LXOR, lxor_func, LAND_TYPES)
255 CREATE_MPI_OP(BAND, band_func, BAND_TYPES)
256 CREATE_MPI_OP(BOR, bor_func, BAND_TYPES)
257 CREATE_MPI_OP(BXOR, bxor_func, BAND_TYPES)
258 CREATE_MPI_OP(MAXLOC, maxloc_func, DT_FLAG_REDUCTION)
259 CREATE_MPI_OP(MINLOC, minloc_func, DT_FLAG_REDUCTION)
260 CREATE_MPI_OP(REPLACE, replace_func, 0)
261 CREATE_MPI_OP(NO_OP, no_func, 0)
263 namespace simgrid::smpi {
265 void Op::apply(const void* invec, void* inoutvec, const int* len, MPI_Datatype datatype) const
267 // we need to switch as the called function may silently touch global variables
268 smpi_switch_data_segment(simgrid::s4u::Actor::self());
270 if (not smpi_process()->replaying() && *len > 0) {
271 XBT_DEBUG("Applying operation of length %d from %p and from/to %p", *len, invec, inoutvec);
272 if (not is_fortran_op_)
273 this->func_(const_cast<void*>(invec), inoutvec, const_cast<int*>(len), &datatype);
275 int tmp = datatype->c2f();
276 /* Unfortunately, the C and Fortran version of the MPI standard do not agree on the type here,
277 thus the reinterpret_cast. */
278 this->func_(const_cast<void*>(invec), inoutvec, const_cast<int*>(len), reinterpret_cast<MPI_Datatype*>(&tmp));
284 return static_cast<Op*>(F2C::f2c(id));
291 void Op::unref(MPI_Op* op){
292 if((*op)!=MPI_OP_NULL){
294 if ((*op)->refcount_ == 0 && not (*op)->is_predefined_){
295 F2C::free_f((*op)->f2c_id());
301 } // namespace simgrid::smpi