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
|
# ---
cdef class _p_greq:
cdef object query_fn
cdef object free_fn
cdef object cancel_fn
cdef object args, kargs
def __cinit__(self, query_fn, free_fn, cancel_fn,
args=None, kargs=None):
self.query_fn = query_fn
self.free_fn = free_fn
self.cancel_fn = cancel_fn
self.args = tuple(args) if args is not None else ()
self.kargs = dict(kargs) if kargs is not None else {}
cdef int query(self, MPI_Status *status) except -1:
cdef Status sts = Status()
if self.query_fn is not None:
self.query_fn(sts, *self.args, **self.kargs)
if (status != NULL and
status != MPI_STATUS_IGNORE and
status != MPI_STATUSES_IGNORE): # just in case ...
status[0] = sts.ob_mpi
return MPI_SUCCESS
cdef int free(self) except -1:
if self.free_fn is not None:
self.free_fn(*self.args, **self.kargs)
return MPI_SUCCESS
cdef int cancel(self, bint completed) except -1:
if self.cancel_fn is not None:
self.cancel_fn(completed, *self.args, **self.kargs)
elif not completed:
return MPI_ERR_PENDING
return MPI_SUCCESS
# ---
cdef int greq_query(void *extra_state, MPI_Status *status) with gil:
cdef int ierr = MPI_SUCCESS
cdef _p_greq state = <_p_greq>extra_state
try:
ierr = state.query(status)
except Exception, exc:
ierr = exc.Get_error_code()
except:
ierr = MPI_ERR_OTHER
return ierr
cdef int greq_free(void *extra_state) with gil:
cdef int ierr = MPI_SUCCESS
cdef _p_greq state = <object>extra_state
try:
ierr = state.free()
except Exception, exc:
ierr = exc.Get_error_code()
except:
ierr = MPI_ERR_OTHER
Py_DECREF(<object>extra_state)
return ierr
cdef int greq_cancel(void *extra_state, int completed) with gil:
cdef int ierr = MPI_SUCCESS
cdef _p_greq state = <object>extra_state
try:
ierr = state.cancel(completed)
except Exception, exc:
ierr = exc.Get_error_code()
except:
ierr = MPI_ERR_OTHER
return ierr
# ---
cdef int greq_query_fn(void *extra_state, MPI_Status *status) nogil:
if Py_IsInitialized():
if extra_state == NULL: return MPI_ERR_INTERN
return greq_query(extra_state, status)
if status != MPI_STATUS_IGNORE: # just in case ...
status.MPI_SOURCE = MPI_ANY_SOURCE
status.MPI_TAG = MPI_ANY_TAG
status.MPI_ERROR = MPI_SUCCESS
MPI_Status_set_elements(status, MPI_BYTE, 0)
MPI_Status_set_cancelled(status, 1)
return MPI_SUCCESS # XXX or MPI_ERR_OTHER ?
cdef int greq_free_fn(void *extra_state) nogil:
if Py_IsInitialized():
if extra_state == NULL: return MPI_ERR_INTERN
return greq_free(extra_state)
return MPI_SUCCESS # XXX or MPI_ERR_OTHER ?
cdef int greq_cancel_fn(void *extra_state, int completed) nogil:
if Py_IsInitialized():
if extra_state == NULL: return MPI_ERR_INTERN
return greq_cancel(extra_state, completed)
return MPI_SUCCESS # XXX or MPI_ERR_OTHER ?
# ---
|
