numba_mpi.api.initialized

file contains MPI_Initialized() implementation

 1"""file contains MPI_Initialized() implementation"""
 2
 3import ctypes
 4
 5import numba
 6import numpy as np
 7
 8from numba_mpi.common import libmpi
 9
10_MPI_Initialized = libmpi.MPI_Initialized
11_MPI_Initialized.restype = ctypes.c_int
12_MPI_Initialized.argtypes = [ctypes.c_void_p]
13
14
15@numba.njit()
16def initialized():
17    """wrapper for MPI_Initialized()"""
18    flag = np.empty((1,), dtype=np.intc)
19    status = _MPI_Initialized(flag.ctypes.data)
20    if status != 0:
21        return False
22    return bool(flag[0])
@numba.njit()
def initialized():
16@numba.njit()
17def initialized():
18    """wrapper for MPI_Initialized()"""
19    flag = np.empty((1,), dtype=np.intc)
20    status = _MPI_Initialized(flag.ctypes.data)
21    if status != 0:
22        return False
23    return bool(flag[0])

wrapper for MPI_Initialized()