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
|
#define Py_LIMITED_API 0x03060000
#define MPI4PY_LIMITED_API 1
#define MPI4PY_LIMITED_API_SKIP_MESSAGE 1
#define MPI4PY_LIMITED_API_SKIP_SESSION 1
#define MPICH_SKIP_MPICXX 1
#define OMPI_SKIP_MPICXX 1
#include <Python.h>
#include <mpi4py/mpi4py.h>
/* -------------------------------------------------------------------------- */
static void
sayhello(MPI_Comm comm) {
int size, rank;
char pname[MPI_MAX_PROCESSOR_NAME]; int len;
if (comm == MPI_COMM_NULL) {
printf("You passed MPI_COMM_NULL !!!\n");
return;
}
MPI_Comm_size(comm, &size);
MPI_Comm_rank(comm, &rank);
MPI_Get_processor_name(pname, &len);
pname[len] = 0;
printf("Hello, World! I am process %d of %d on %s.\n",
rank, size, pname);
}
/* -------------------------------------------------------------------------- */
static PyObject *
hw_sayhello(PyObject *self, PyObject *args)
{
PyObject *py_comm = NULL;
MPI_Comm *comm_p = NULL;
if (!PyArg_ParseTuple(args, "O:sayhello", &py_comm))
return NULL;
comm_p = PyMPIComm_Get(py_comm);
if (comm_p == NULL)
return NULL;
sayhello(*comm_p);
Py_INCREF(Py_None);
return Py_None;
}
static struct PyMethodDef hw_methods[] = {
{"sayhello", (PyCFunction)hw_sayhello, METH_VARARGS, NULL},
{NULL, NULL, 0, NULL} /* sentinel */
};
static struct PyModuleDef hw_module = {
PyModuleDef_HEAD_INIT,
"helloworld", /* m_name */
NULL, /* m_doc */
-1, /* m_size */
hw_methods /* m_methods */,
NULL, /* m_reload */
NULL, /* m_traverse */
NULL, /* m_clear */
NULL /* m_free */
};
PyMODINIT_FUNC PyInit_helloworld(void);
PyMODINIT_FUNC PyInit_helloworld(void)
{
PyObject *m = NULL;
/* Initialize mpi4py's C-API */
if (import_mpi4py() < 0) goto bad;
/* Module initialization */
m = PyModule_Create(&hw_module);
if (m == NULL) goto bad;
return m;
bad:
return NULL;
}
/* -------------------------------------------------------------------------- */
/*
Local variables:
c-basic-offset: 2
indent-tabs-mode: nil
End:
*/
|
