numba_mpi.api.barrier

file contains MPI_Barrier() implementation

View Source

 1"""file contains MPI_Barrier() implementation"""
 2
 3import ctypes
 4
 5import numba
 6
 7from numba_mpi.common import _MPI_Comm_World_ptr, libmpi
 8from numba_mpi.utils import _mpi_addr, _MpiComm
 9
10_MPI_Barrier = libmpi.MPI_Barrier
11_MPI_Barrier.restype = ctypes.c_int
12_MPI_Barrier.argtypes = [_MpiComm]
13
14
15@numba.njit()
16def barrier():
17    """wrapper for MPI_Barrier(). Returns integer status code (0 == MPI_SUCCESS)"""
18    return _MPI_Barrier(_mpi_addr(_MPI_Comm_World_ptr))

@numba.njit()

def barrier(): View Source

16@numba.njit()
17def barrier():
18    """wrapper for MPI_Barrier(). Returns integer status code (0 == MPI_SUCCESS)"""
19    return _MPI_Barrier(_mpi_addr(_MPI_Comm_World_ptr))

wrapper for MPI_Barrier(). Returns integer status code (0 == MPI_SUCCESS)