File: helloworld.pyx

package info (click to toggle)
mpi4py 2.0.0-2.1%2Bdeb9u1
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 2,680 kB
  • sloc: python: 15,291; ansic: 7,099; makefile: 719; f90: 158; sh: 156; cpp: 121
file content (23 lines) | stat: -rw-r--r-- 654 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
cdef extern from "mpi-compat.h": pass

cimport mpi4py.MPI as MPI
from mpi4py.libmpi cimport *

cdef extern from "stdio.h":
    int printf(char*, ...)

cdef void c_sayhello(MPI_Comm comm):
    cdef int size, rank, plen
    cdef char pname[MPI_MAX_PROCESSOR_NAME]
    if comm == MPI_COMM_NULL:
        printf(b"You passed MPI_COMM_NULL !!!\n",0)
        return
    MPI_Comm_size(comm, &size)
    MPI_Comm_rank(comm, &rank)
    MPI_Get_processor_name(pname, &plen)
    printf(b"Hello, World! I am process %d of %d on %s.\n",
           rank, size, pname)

def sayhello(MPI.Comm comm not None ):
    cdef MPI_Comm c_comm = comm.ob_mpi
    c_sayhello(c_comm)