Package numba_mpi

Numba @njittable MPI wrappers tested on Linux, macOS and Windows

Expand source code
""" Numba @njittable MPI wrappers tested on Linux, macOS and Windows """

from importlib.metadata import PackageNotFoundError, version

from .api.allreduce import allreduce
from .api.barrier import barrier
from .api.bcast import bcast
from .api.initialized import initialized
from .api.irecv import irecv
from .api.isend import isend
from .api.operator import Operator
from .api.rank import rank
from .api.recv import recv
from .api.requests import test, testall, testany, wait, waitall, waitany
from .api.scatter_gather import allgather, gather, scatter
from .api.send import send
from .api.size import size
from .api.wtime import wtime
from .common import RequestType

try:
    __version__ = version(__name__)
except PackageNotFoundError:
    # package is not installed
    pass

Sub-modules

numba_mpi.api

functions that are part of the package API

numba_mpi.common

variables used across API implementation

numba_mpi.utils

helper functions used across API implementation