numba-mpi 0.18

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

numba-mpi

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

API documentation is maintained at: https://atmos-cloud-sim-uj.github.io/numba-mpi

Hello world example:

import numba, numba_mpi, numpy

@numba.njit()
def hello():
  print(numba_mpi.rank())
  print(numba_mpi.size())

  src = numpy.array([1., 2., 3., 4., 5.])
  dst_tst = numpy.empty_like(src)

  if numba_mpi.rank() == 0:
    numba_mpi.send(src, dest=1, tag=11)
  elif numba_mpi.rank() == 1:
    numba_mpi.recv(dst_tst, source=0, tag=11)

hello()

For information on MPI, see:

• https://www.mpi-forum.org/docs/
• https://www.open-mpi.org/doc/
• https://www.mpich.org/
• https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi
• https://mpi4py.readthedocs.io/en/stable/