numba_mpi.api.query_thread

file contains MPI_Query_thread() implementation

View Source

 1"""file contains MPI_Query_thread() implementation"""
 2
 3import ctypes
 4
 5import numba
 6import numpy as np
 7
 8from numba_mpi.common import libmpi
 9
10_MPI_Query_thread = libmpi.MPI_Query_thread
11_MPI_Query_thread.restype = ctypes.c_int
12_MPI_Query_thread.argtypes = [ctypes.c_void_p]
13
14
15@numba.njit()
16def query_thread():
17    """wrapper for MPI_Query_thread()"""
18    provided = np.empty(1, dtype=np.intc)
19    _ = _MPI_Query_thread(provided.ctypes.data)
20    return provided[0]

@numba.njit()

def query_thread(): View Source

16@numba.njit()
17def query_thread():
18    """wrapper for MPI_Query_thread()"""
19    provided = np.empty(1, dtype=np.intc)
20    _ = _MPI_Query_thread(provided.ctypes.data)
21    return provided[0]

wrapper for MPI_Query_thread()