1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
|
#---------------------------------------------------------------------
cdef enum PyMPI_OBJECT_FLAGS:
PyMPI_OWNED = 1<<1
#---------------------------------------------------------------------
# Status
cdef inline MPI_Status *_arg_Status(object status):
if status is None: return MPI_STATUS_IGNORE
return &((<Status>status).ob_mpi)
cdef inline int _eq_Status(MPI_Status* s1, MPI_Status* s2) nogil:
cdef size_t i=0, n=sizeof(MPI_Status)
cdef unsigned char* a = <unsigned char*>s1
cdef unsigned char* b = <unsigned char*>s2
for i from 0 <= i < n:
if a[i] != b[i]:
return 0
return 1
#---------------------------------------------------------------------
# Datatype
cdef inline Datatype _new_Datatype(MPI_Datatype ob):
cdef Datatype datatype = Datatype()
datatype.ob_mpi = ob
return datatype
cdef inline int _del_Datatype(MPI_Datatype* ob):
#
if ob == NULL : return 0
if ob[0] == MPI_DATATYPE_NULL : return 0
if not mpi_active() : return 0
#
return MPI_Type_free(ob)
cdef inline int _named_Datatype(MPI_Datatype ob):
if ob == MPI_DATATYPE_NULL: return 0
cdef int ni = 0, na = 0, nt = 0, combiner = MPI_UNDEFINED
cdef int ierr = MPI_SUCCESS
ierr = MPI_Type_get_envelope(ob, &ni, &na, &nt, &combiner)
if ierr: return 0 # XXX
return combiner == MPI_COMBINER_NAMED
cdef void _fix_Datatype(Datatype datatype):
cdef MPI_Datatype ob = datatype.ob_mpi
if ob == MPI_DATATYPE_NULL: return
if _named_Datatype(ob): pass
#---------------------------------------------------------------------
# Request
include "reqimpl.pxi"
cdef inline Request _new_Request(MPI_Request ob):
cdef Request request = Request()
request.ob_mpi = ob
return request
cdef inline int _del_Request(MPI_Request* ob):
#
if ob == NULL : return 0
if ob[0] == MPI_REQUEST_NULL : return 0
if not mpi_active() : return 0
#
return MPI_Request_free(ob)
#---------------------------------------------------------------------
# Op
include "opimpl.pxi"
cdef inline Op _new_Op(MPI_Op ob):
cdef Op op = Op()
op.ob_mpi = ob
if ob == MPI_OP_NULL : op.ob_func = NULL
elif ob == MPI_MAX : op.ob_func = _op_MAX
elif ob == MPI_MIN : op.ob_func = _op_MIN
elif ob == MPI_SUM : op.ob_func = _op_SUM
elif ob == MPI_PROD : op.ob_func = _op_PROD
elif ob == MPI_LAND : op.ob_func = _op_LAND
elif ob == MPI_BAND : op.ob_func = _op_BAND
elif ob == MPI_LOR : op.ob_func = _op_LOR
elif ob == MPI_BOR : op.ob_func = _op_BOR
elif ob == MPI_LXOR : op.ob_func = _op_LXOR
elif ob == MPI_BXOR : op.ob_func = _op_BXOR
elif ob == MPI_MAXLOC : op.ob_func = _op_MAXLOC
elif ob == MPI_MINLOC : op.ob_func = _op_MINLOC
elif ob == MPI_REPLACE : op.ob_func = _op_REPLACE
return op
cdef inline int _del_Op(MPI_Op* ob):
#
if ob == NULL : return 0
if ob[0] == MPI_OP_NULL : return 0
if ob[0] == MPI_MAX : return 0
if ob[0] == MPI_MIN : return 0
if ob[0] == MPI_SUM : return 0
if ob[0] == MPI_PROD : return 0
if ob[0] == MPI_LAND : return 0
if ob[0] == MPI_BAND : return 0
if ob[0] == MPI_LOR : return 0
if ob[0] == MPI_BOR : return 0
if ob[0] == MPI_LXOR : return 0
if ob[0] == MPI_BXOR : return 0
if ob[0] == MPI_MAXLOC : return 0
if ob[0] == MPI_MINLOC : return 0
if ob[0] == MPI_REPLACE : return 0
if not mpi_active() : return 0
#
return MPI_Op_free(ob)
#---------------------------------------------------------------------
# Info
cdef inline Info _new_Info(MPI_Info ob):
cdef Info info = Info()
info.ob_mpi = ob
return info
cdef inline int _del_Info(MPI_Info* ob):
#
if ob == NULL : return 0
if ob[0] == MPI_INFO_NULL : return 0
if not mpi_active() : return 0
#
return MPI_Info_free(ob)
cdef inline MPI_Info _arg_Info(object info):
if info is None: return MPI_INFO_NULL
return (<Info>info).ob_mpi
#---------------------------------------------------------------------
# Group
cdef inline Group _new_Group(MPI_Group ob):
cdef Group group = Group()
group.ob_mpi = ob
return group
cdef inline int _del_Group(MPI_Group* ob):
#
if ob == NULL : return 0
if ob[0] == MPI_GROUP_NULL : return 0
if ob[0] == MPI_GROUP_EMPTY : return 0
if not mpi_active() : return 0
#
return MPI_Group_free(ob)
#---------------------------------------------------------------------
# Comm
cdef inline Comm _new_Comm(MPI_Comm ob):
cdef Comm comm = Comm()
comm.ob_mpi = ob
return comm
cdef inline Intracomm _new_Intracomm(MPI_Comm ob):
cdef Intracomm comm = Intracomm()
comm.ob_mpi = ob
return comm
cdef inline Intercomm _new_Intercomm(MPI_Comm ob):
cdef Intercomm comm = Intercomm()
comm.ob_mpi = ob
return comm
cdef inline int _del_Comm(MPI_Comm* ob):
#
if ob == NULL : return 0
if ob[0] == MPI_COMM_NULL : return 0
if ob[0] == MPI_COMM_SELF : return 0
if ob[0] == MPI_COMM_WORLD : return 0
if not mpi_active() : return 0
#
return MPI_Comm_free(ob)
#---------------------------------------------------------------------
# Win
include "winimpl.pxi"
cdef inline Win _new_Win(MPI_Win ob):
cdef Win win = Win()
win.ob_mpi = ob
return win
cdef inline int _del_Win(MPI_Win* ob):
#
if ob == NULL : return 0
if ob[0] == MPI_WIN_NULL : return 0
if not mpi_active() : return 0
#
return MPI_Win_free(ob)
#---------------------------------------------------------------------
# File
cdef inline File _new_File(MPI_File ob):
cdef File file = File()
file.ob_mpi = ob
return file
cdef inline int _del_File(MPI_File* ob):
#
if ob == NULL : return 0
if ob[0] == MPI_FILE_NULL : return 0
if not mpi_active() : return 0
#
return MPI_File_close(ob)
#---------------------------------------------------------------------
# Errhandler
cdef inline Errhandler _new_Errhandler(MPI_Errhandler ob):
cdef Errhandler errhandler = Errhandler()
errhandler.ob_mpi = ob
return errhandler
cdef inline int _del_Errhandler(MPI_Errhandler* ob):
#
if ob == NULL : return 0
if ob[0] == MPI_ERRHANDLER_NULL : return 0
if not mpi_active() : return 0
#
return MPI_Errhandler_free(ob)
#---------------------------------------------------------------------
|
