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()