numba-mpi 0.39__tar.gz → 0.41__tar.gz

{numba-mpi-0.39 → numba_mpi-0.41}/.github/workflows/readme_snippets.yml

@@ -22,7 +22,17 @@ jobs:
           python-version: "3.10"
       - uses: mpi4py/setup-mpi@v1
       - run: pip install -e .
-      - run: pip install pytest-codeblocks pytest
+      - run: pip install pytest-codeblocks pytest matplotlib
       - run: python -c "import pytest_codeblocks; code=pytest_codeblocks.extract_from_file('README.md'); f=open('readme.py', 'w'); f.writelines(block.code for block in code if block.syntax=='python'); f.close()"
       - run: cat -n readme.py
-      - run: mpiexec -n 2 python -We readme.py
+      - run: mpiexec -n 4 python -We readme.py
+      - uses: actions/upload-artifact@v3
+        with:
+          path: readme_plot.png
+          name: readme_plot
+      - if: github.ref == 'refs/heads/main'
+        uses: eine/tip@master
+        with:
+          token: ${{ secrets.GITHUB_TOKEN }}
+          files: |
+            readme_plot.png

{numba-mpi-0.39 → numba_mpi-0.41}/.github/workflows/stale.yml

@@ -7,7 +7,7 @@ name: Mark stale issues and pull requests
 
 on:
   schedule:
-  - cron: '33 12 * * *'
+  - cron: '45 12 * * *'
 
 jobs:
   stale:
@@ -18,10 +18,13 @@ jobs:
       pull-requests: write
 
     steps:
-    - uses: actions/stale@v3
+    - uses: actions/stale@v9
       with:
         repo-token: ${{ secrets.GITHUB_TOKEN }}
         stale-issue-message: 'Stale issue message'
         stale-pr-message: 'Stale pull request message'
-        stale-issue-label: 'no-issue-activity'
-        stale-pr-label: 'no-pr-activity'
+        stale-issue-label: 'no-activity'
+        stale-pr-label: 'no-activity'
+        operations-per-run: 3
+        days-before-stale: 90
+        days-before-close: 30

{numba-mpi-0.39 → numba_mpi-0.41}/.github/workflows/tests+pypi.yml

@@ -27,9 +27,10 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        pip install pylint==2.14.5 pytest
+        pip install pylint==2.14.5
         python setup.py egg_info
-        pip install -r *.egg-info/requires.txt
+        ex -sc 'g/^\[tests\]$/d' -cx numba_mpi.egg-info/requires.txt
+        pip install -r numba_mpi.egg-info/requires.txt
     - name: Analysing the code with pylint
       run: |
         pylint --unsafe-load-any-extension=y --disable=fixme $(git ls-files '*.py')
@@ -87,13 +88,13 @@ jobs:
     needs: [pylint, precommit, pdoc, zenodo_json]
     strategy:
       matrix:
-        platform: [ubuntu-latest, macos-latest, windows-latest]
-        python-version: ["3.7", "3.8", "3.9", "3.10", "3.11"]
+        platform: [ubuntu-latest, macos-12, windows-latest]
+        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
         mpi: [ 'mpich', 'openmpi', 'msmpi', 'intelmpi']
         exclude:
-        - platform: macos-latest
+        - platform: macos-12
           mpi: msmpi
-        - platform: macos-latest
+        - platform: macos-12
           mpi: intelmpi
         - platform: ubuntu-latest
           mpi: msmpi
@@ -104,10 +105,7 @@ jobs:
         - platform: windows-latest
           mpi: intelmpi
           
-        # https://github.com/atmos-cloud-sim-uj/numba-mpi/issues/69
-        - platform: ubuntu-latest
-          python-version: 3.7
-          mpi: mpich
+        # https://github.com/numba-mpi/numba-mpi/issues/69
         - platform: ubuntu-latest
           python-version: 3.8
           mpi: mpich
@@ -129,8 +127,7 @@ jobs:
           # checking if multiple MPI libs are OK
           sudo apt-get install mpi-default-bin
           mpiexec python -c "import os; assert 'I_MPI_ROOT' in os.environ"
-      - run: pip install -e .
-      - run: pip install pytest
+      - run: pip install -e .[tests]
       - run: python -We -c "import mpi4py"
       - run: python -We -c "import numba_mpi"
       - run: mpiexec -n 2 pytest --durations=10 -p no:unraisableexception -We
@@ -160,7 +157,8 @@ jobs:
 
       - run: |
           python setup.py egg_info
-          pip install -r *.egg-info/requires.txt
+          ex -sc 'g/^\[tests\]$/d' -cx numba_mpi.egg-info/requires.txt
+          pip install -r numba_mpi.egg-info/requires.txt
           pip install --no-index --find-links file:./dist numba-mpi
           cd /tmp  # make sure local files are not picked up
           python -We -c "import numba_mpi"

{numba-mpi-0.39 → numba_mpi-0.41}/.pre-commit-config.yaml

@@ -4,18 +4,18 @@ default_stages: [commit]
 
 repos:
   - repo: https://github.com/psf/black
-    rev: 23.1.0
+    rev: 24.1.1
     hooks:
       - id: black
 
   - repo: https://github.com/timothycrosley/isort
-    rev: 5.12.0
+    rev: 5.13.2
     hooks:
       - id: isort
         args: ["--profile", "black"]
 
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v4.4.0
+    rev: v4.5.0
     hooks:
       - id: trailing-whitespace
       - id: end-of-file-fixer

numba_mpi-0.41/PKG-INFO

@@ -0,0 +1,172 @@
+Metadata-Version: 2.1
+Name: numba-mpi
+Version: 0.41
+Summary: Numba @njittable MPI wrappers tested on Linux, macOS and Windows
+Home-page: https://github.com/numba-mpi/numba-mpi
+Author: https://github.com/numba-mpi/numba-mpi/graphs/contributors
+License: GPL v3
+Project-URL: Tracker, https://github.com/numba-mpi/numba-mpi/issues
+Project-URL: Documentation, https://numba-mpi.github.io/numba-mpi
+Project-URL: Source, https://github.com/numba-mpi/numba-mpi
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numba
+Requires-Dist: numpy
+Requires-Dist: mpi4py
+Requires-Dist: psutil
+Provides-Extra: tests
+Requires-Dist: pytest<8.0.0; extra == "tests"
+
+# <img src="https://raw.githubusercontent.com/numba-mpi/numba-mpi/main/.github/numba_mpi_logo.svg" style="height:50pt" alt="numba-mpi logo"> numba-mpi
+
+[![Python 3](https://img.shields.io/static/v1?label=Python&logo=Python&color=3776AB&message=3)](https://www.python.org/)
+[![LLVM](https://img.shields.io/static/v1?label=LLVM&logo=LLVM&color=gold&message=Numba)](https://numba.pydata.org)
+[![Linux OK](https://img.shields.io/static/v1?label=Linux&logo=Linux&color=yellow&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Linux)
+[![macOS OK](https://img.shields.io/static/v1?label=macOS&logo=Apple&color=silver&message=%E2%9C%93)](https://en.wikipedia.org/wiki/macOS)
+[![Windows OK](https://img.shields.io/static/v1?label=Windows&logo=Windows&color=white&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Windows)
+[![Github Actions Status](https://github.com/numba-mpi/numba-mpi/workflows/tests+pypi/badge.svg?branch=main)](https://github.com/numba-mpi/numba-mpi/actions/workflows/tests+pypi.yml)
+[![Maintenance](https://img.shields.io/badge/Maintained%3F-yes-green.svg)](https://GitHub.com/numba-mpi/numba-mpi/graphs/commit-activity)
+[![License: GPL v3](https://img.shields.io/badge/License-GPL%20v3-blue.svg)](https://www.gnu.org/licenses/gpl-3.0.html)
+[![PyPI version](https://badge.fury.io/py/numba-mpi.svg)](https://pypi.org/project/numba-mpi)
+[![Anaconda-Server Badge](https://anaconda.org/conda-forge/numba-mpi/badges/version.svg)](https://anaconda.org/conda-forge/numba-mpi)
+[![AUR package](https://repology.org/badge/version-for-repo/aur/python:numba-mpi.svg)](https://aur.archlinux.org/packages/python-numba-mpi)
+[![DOI](https://zenodo.org/badge/316911228.svg)](https://zenodo.org/badge/latestdoi/316911228)
+
+### Overview
+numba-mpi provides Python wrappers to the C MPI API callable from within [Numba JIT-compiled code](https://numba.readthedocs.io/en/stable/user/jit.html) (@njit mode).
+
+Support is provided for a subset of MPI routines covering: `size`/`rank`, `send`/`recv`, `allreduce`, `bcast`, `scatter`/`gather` & `allgather`, `barrier`, `wtime`
+and basic asynchronous communication with `isend`/`irecv` (only for contiguous arrays); for request handling including `wait`/`waitall`/`waitany` and `test`/`testall`/`testany`.
+
+The API uses NumPy and supports both numeric and character datatypes (e.g., `broadcast`). 
+Auto-generated docstring-based API docs are published on the web: https://numba-mpi.github.io/numba-mpi
+
+Packages can be obtained from 
+  [PyPI](https://pypi.org/project/numba-mpi), 
+  [Conda Forge](https://anaconda.org/conda-forge/numba-mpi), 
+  [Arch Linux](https://aur.archlinux.org/packages/python-numba-mpi)
+  or by invoking `pip install git+https://github.com/numba-mpi/numba-mpi.git`.
+
+numba-mpi is a pure-Python package.
+The codebase includes a test suite used through the GitHub Actions workflows ([thanks to mpi4py's setup-mpi](https://github.com/mpi4py/setup-mpi)!)
+for automated testing on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/) 
+& [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)), 
+macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and 
+Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi)).
+
+Features that are not implemented yet include (help welcome!):
+- support for non-default communicators
+- support for `MPI_IN_PLACE` in `[all]gather`/`scatter` and `allreduce`
+- support for `MPI_Type_create_struct` (Numpy structured arrays) 
+- ...
+
+### Hello world send/recv example:
+```python
+import numba, numba_mpi, numpy
+
+@numba.njit()
+def hello():
+    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()
+```
+
+### Example comparing numba-mpi vs. mpi4py performance:
+
+The example below compares Numba + mpi4py vs. Numba + numba-mpi performance.
+The sample code estimates $\pi$ by integration of $4/(1+x^2)$ between 0 and 1
+dividing the workload into `n_intervals` handled by separate MPI processes 
+and then obtaining a sum using `allreduce`.
+The computation is carried out in a JIT-compiled function and is repeated
+`N_TIMES`, the repetitions and the MPI-handled reduction are done outside or 
+inside of the JIT-compiled block for mpi4py and numba-mpi, respectively.
+Timing is repeated `N_REPEAT` times and the minimum time is reported.
+The generated plot shown below depicts the speedup obtained by replacing mpi4py
+with numba_mpi as a function of `n_intervals` - the more often communication
+is needed (smaller `n_intervals`), the larger the expected speedup.
+
+```python
+import timeit, mpi4py, numba, numpy as np, numba_mpi
+
+N_TIMES = 10000
+N_REPEAT = 10
+RTOL = 1e-3
+
+@numba.njit
+def get_pi_part(out, n_intervals, rank, size):
+    h = 1 / n_intervals
+    partial_sum = 0.0
+    for i in range(rank + 1, n_intervals, size):
+        x = h * (i - 0.5)
+        partial_sum += 4 / (1 + x**2)
+    out[0] = h * partial_sum
+
+@numba.njit
+def pi_numba_mpi(n_intervals):
+    pi = np.array([0.])
+    part = np.empty_like(pi)
+    for _ in range(N_TIMES):
+        get_pi_part(part, n_intervals, numba_mpi.rank(), numba_mpi.size())
+        numba_mpi.allreduce(part, pi, numba_mpi.Operator.SUM)
+        assert abs(pi[0] - np.pi) / np.pi < RTOL
+
+def pi_mpi4py(n_intervals):
+    pi = np.array([0.])
+    part = np.empty_like(pi)
+    for _ in range(N_TIMES):
+        get_pi_part(part, n_intervals, mpi4py.MPI.COMM_WORLD.rank, mpi4py.MPI.COMM_WORLD.size)
+        mpi4py.MPI.COMM_WORLD.Allreduce(part, (pi, mpi4py.MPI.DOUBLE), op=mpi4py.MPI.SUM)
+        assert abs(pi[0] - np.pi) / np.pi < RTOL
+
+plot_x = [1000 * k for k in range(1, 11)]
+plot_y = {'numba_mpi': [], 'mpi4py': []}
+for n_intervals in plot_x:
+    for impl in plot_y:
+        plot_y[impl].append(min(timeit.repeat(
+            f"pi_{impl}({n_intervals})",
+            globals=locals(),
+            number=1,
+            repeat=N_REPEAT
+        )))
+
+if numba_mpi.rank() == 0:
+    from matplotlib import pyplot
+    pyplot.figure(figsize=(8.3, 3.5), tight_layout=True)
+    pyplot.plot(plot_x, np.array(plot_y['mpi4py'])/np.array(plot_y['numba_mpi']), marker='o')
+    pyplot.xlabel('n_intervals (workload in between communication)')
+    pyplot.ylabel('wall time ratio (mpi4py / numba_mpi)')
+    pyplot.title(f'mpiexec -np {numba_mpi.size()}')
+    pyplot.grid()
+    pyplot.savefig('readme_plot.png')
+```
+
+![plot](https://github.com/numba-mpi/numba-mpi/releases/download/tip/readme_plot.png)
+
+
+### MPI resources on the web:
+
+- MPI standard and general information:
+    - https://www.mpi-forum.org/docs
+    - https://en.wikipedia.org/wiki/Message_Passing_Interface
+- MPI implementations:
+    - OpenMPI: https://www.open-mpi.org
+    - MPICH: https://www.mpich.org
+    - MS MPI: https://learn.microsoft.com/en-us/message-passing-interface
+    - Intel MPI: https://intel.com/content/www/us/en/developer/tools/oneapi/mpi-library-documentation.html
+- MPI bindings:
+    - Python: https://mpi4py.readthedocs.io
+    - Julia: https://juliaparallel.org/MPI.jl
+    - Rust: https://docs.rs/mpi
+    - C++: https://boost.org/doc/html/mpi.html
+    - R: https://cran.r-project.org/web/packages/Rmpi
+
+### Acknowledgements:
+
+Development of numba-mpi has been supported by the [Polish National Science Centre](https://ncn.gov.pl/en) (grant no. 2020/39/D/ST10/01220).
+

numba_mpi-0.41/README.md

@@ -0,0 +1,153 @@
+# <img src="https://raw.githubusercontent.com/numba-mpi/numba-mpi/main/.github/numba_mpi_logo.svg" style="height:50pt" alt="numba-mpi logo"> numba-mpi
+
+[![Python 3](https://img.shields.io/static/v1?label=Python&logo=Python&color=3776AB&message=3)](https://www.python.org/)
+[![LLVM](https://img.shields.io/static/v1?label=LLVM&logo=LLVM&color=gold&message=Numba)](https://numba.pydata.org)
+[![Linux OK](https://img.shields.io/static/v1?label=Linux&logo=Linux&color=yellow&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Linux)
+[![macOS OK](https://img.shields.io/static/v1?label=macOS&logo=Apple&color=silver&message=%E2%9C%93)](https://en.wikipedia.org/wiki/macOS)
+[![Windows OK](https://img.shields.io/static/v1?label=Windows&logo=Windows&color=white&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Windows)
+[![Github Actions Status](https://github.com/numba-mpi/numba-mpi/workflows/tests+pypi/badge.svg?branch=main)](https://github.com/numba-mpi/numba-mpi/actions/workflows/tests+pypi.yml)
+[![Maintenance](https://img.shields.io/badge/Maintained%3F-yes-green.svg)](https://GitHub.com/numba-mpi/numba-mpi/graphs/commit-activity)
+[![License: GPL v3](https://img.shields.io/badge/License-GPL%20v3-blue.svg)](https://www.gnu.org/licenses/gpl-3.0.html)
+[![PyPI version](https://badge.fury.io/py/numba-mpi.svg)](https://pypi.org/project/numba-mpi)
+[![Anaconda-Server Badge](https://anaconda.org/conda-forge/numba-mpi/badges/version.svg)](https://anaconda.org/conda-forge/numba-mpi)
+[![AUR package](https://repology.org/badge/version-for-repo/aur/python:numba-mpi.svg)](https://aur.archlinux.org/packages/python-numba-mpi)
+[![DOI](https://zenodo.org/badge/316911228.svg)](https://zenodo.org/badge/latestdoi/316911228)
+
+### Overview
+numba-mpi provides Python wrappers to the C MPI API callable from within [Numba JIT-compiled code](https://numba.readthedocs.io/en/stable/user/jit.html) (@njit mode).
+
+Support is provided for a subset of MPI routines covering: `size`/`rank`, `send`/`recv`, `allreduce`, `bcast`, `scatter`/`gather` & `allgather`, `barrier`, `wtime`
+and basic asynchronous communication with `isend`/`irecv` (only for contiguous arrays); for request handling including `wait`/`waitall`/`waitany` and `test`/`testall`/`testany`.
+
+The API uses NumPy and supports both numeric and character datatypes (e.g., `broadcast`). 
+Auto-generated docstring-based API docs are published on the web: https://numba-mpi.github.io/numba-mpi
+
+Packages can be obtained from 
+  [PyPI](https://pypi.org/project/numba-mpi), 
+  [Conda Forge](https://anaconda.org/conda-forge/numba-mpi), 
+  [Arch Linux](https://aur.archlinux.org/packages/python-numba-mpi)
+  or by invoking `pip install git+https://github.com/numba-mpi/numba-mpi.git`.
+
+numba-mpi is a pure-Python package.
+The codebase includes a test suite used through the GitHub Actions workflows ([thanks to mpi4py's setup-mpi](https://github.com/mpi4py/setup-mpi)!)
+for automated testing on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/) 
+& [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)), 
+macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and 
+Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi)).
+
+Features that are not implemented yet include (help welcome!):
+- support for non-default communicators
+- support for `MPI_IN_PLACE` in `[all]gather`/`scatter` and `allreduce`
+- support for `MPI_Type_create_struct` (Numpy structured arrays) 
+- ...
+
+### Hello world send/recv example:
+```python
+import numba, numba_mpi, numpy
+
+@numba.njit()
+def hello():
+    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()
+```
+
+### Example comparing numba-mpi vs. mpi4py performance:
+
+The example below compares Numba + mpi4py vs. Numba + numba-mpi performance.
+The sample code estimates $\pi$ by integration of $4/(1+x^2)$ between 0 and 1
+dividing the workload into `n_intervals` handled by separate MPI processes 
+and then obtaining a sum using `allreduce`.
+The computation is carried out in a JIT-compiled function and is repeated
+`N_TIMES`, the repetitions and the MPI-handled reduction are done outside or 
+inside of the JIT-compiled block for mpi4py and numba-mpi, respectively.
+Timing is repeated `N_REPEAT` times and the minimum time is reported.
+The generated plot shown below depicts the speedup obtained by replacing mpi4py
+with numba_mpi as a function of `n_intervals` - the more often communication
+is needed (smaller `n_intervals`), the larger the expected speedup.
+
+```python
+import timeit, mpi4py, numba, numpy as np, numba_mpi
+
+N_TIMES = 10000
+N_REPEAT = 10
+RTOL = 1e-3
+
+@numba.njit
+def get_pi_part(out, n_intervals, rank, size):
+    h = 1 / n_intervals
+    partial_sum = 0.0
+    for i in range(rank + 1, n_intervals, size):
+        x = h * (i - 0.5)
+        partial_sum += 4 / (1 + x**2)
+    out[0] = h * partial_sum
+
+@numba.njit
+def pi_numba_mpi(n_intervals):
+    pi = np.array([0.])
+    part = np.empty_like(pi)
+    for _ in range(N_TIMES):
+        get_pi_part(part, n_intervals, numba_mpi.rank(), numba_mpi.size())
+        numba_mpi.allreduce(part, pi, numba_mpi.Operator.SUM)
+        assert abs(pi[0] - np.pi) / np.pi < RTOL
+
+def pi_mpi4py(n_intervals):
+    pi = np.array([0.])
+    part = np.empty_like(pi)
+    for _ in range(N_TIMES):
+        get_pi_part(part, n_intervals, mpi4py.MPI.COMM_WORLD.rank, mpi4py.MPI.COMM_WORLD.size)
+        mpi4py.MPI.COMM_WORLD.Allreduce(part, (pi, mpi4py.MPI.DOUBLE), op=mpi4py.MPI.SUM)
+        assert abs(pi[0] - np.pi) / np.pi < RTOL
+
+plot_x = [1000 * k for k in range(1, 11)]
+plot_y = {'numba_mpi': [], 'mpi4py': []}
+for n_intervals in plot_x:
+    for impl in plot_y:
+        plot_y[impl].append(min(timeit.repeat(
+            f"pi_{impl}({n_intervals})",
+            globals=locals(),
+            number=1,
+            repeat=N_REPEAT
+        )))
+
+if numba_mpi.rank() == 0:
+    from matplotlib import pyplot
+    pyplot.figure(figsize=(8.3, 3.5), tight_layout=True)
+    pyplot.plot(plot_x, np.array(plot_y['mpi4py'])/np.array(plot_y['numba_mpi']), marker='o')
+    pyplot.xlabel('n_intervals (workload in between communication)')
+    pyplot.ylabel('wall time ratio (mpi4py / numba_mpi)')
+    pyplot.title(f'mpiexec -np {numba_mpi.size()}')
+    pyplot.grid()
+    pyplot.savefig('readme_plot.png')
+```
+
+![plot](https://github.com/numba-mpi/numba-mpi/releases/download/tip/readme_plot.png)
+
+
+### MPI resources on the web:
+
+- MPI standard and general information:
+    - https://www.mpi-forum.org/docs
+    - https://en.wikipedia.org/wiki/Message_Passing_Interface
+- MPI implementations:
+    - OpenMPI: https://www.open-mpi.org
+    - MPICH: https://www.mpich.org
+    - MS MPI: https://learn.microsoft.com/en-us/message-passing-interface
+    - Intel MPI: https://intel.com/content/www/us/en/developer/tools/oneapi/mpi-library-documentation.html
+- MPI bindings:
+    - Python: https://mpi4py.readthedocs.io
+    - Julia: https://juliaparallel.org/MPI.jl
+    - Rust: https://docs.rs/mpi
+    - C++: https://boost.org/doc/html/mpi.html
+    - R: https://cran.r-project.org/web/packages/Rmpi
+
+### Acknowledgements:
+
+Development of numba-mpi has been supported by the [Polish National Science Centre](https://ncn.gov.pl/en) (grant no. 2020/39/D/ST10/01220).
+

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/__init__.py

@@ -1,5 +1,6 @@
 """ Numba @njittable MPI wrappers tested on Linux, macOS and Windows """
-from pkg_resources import DistributionNotFound, VersionConflict, get_distribution
+
+from importlib.metadata import PackageNotFoundError, version
 
 from .api.allreduce import allreduce
 from .api.barrier import barrier
@@ -18,7 +19,7 @@ from .api.wtime import wtime
 from .common import RequestType
 
 try:
-    __version__ = get_distribution(__name__).version
-except (DistributionNotFound, VersionConflict):
+    __version__ = version(__name__)
+except PackageNotFoundError:
     # package is not installed
     pass

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/allreduce.py

@@ -1,4 +1,5 @@
 """file contains MPI_Allreduce() implementations"""
+
 import ctypes
 from numbers import Number
 

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/barrier.py

@@ -1,4 +1,5 @@
 """file contains MPI_Barrier() implementation"""
+
 import ctypes
 
 import numba

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/bcast.py

@@ -1,4 +1,5 @@
 """MPI_Bcast() implementation"""
+
 import ctypes
 
 import numba

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/initialized.py

@@ -1,4 +1,5 @@
 """file contains MPI_Initialized() implementation"""
+
 import ctypes
 
 import numba

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/operator.py

@@ -1,4 +1,5 @@
 """operators supported by MPI"""
+
 from enum import IntEnum
 
 from mpi4py import MPI

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/rank.py

@@ -1,4 +1,5 @@
 """MPI_Comm_rank() implementation"""
+
 import ctypes
 
 import numba

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/recv.py

@@ -1,4 +1,5 @@
 """MPI_Recv() implementation"""
+
 import ctypes
 
 import numba

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/requests.py

@@ -1,6 +1,7 @@
 """ Utilities for handling MPI_Requests with respective Wait and Test function
     wrappers.
 """
+
 import ctypes
 
 import numba

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/scatter_gather.py

@@ -1,4 +1,5 @@
 """MPI_Scatter() implementation"""
+
 import ctypes
 
 import numba

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/send.py

@@ -1,4 +1,5 @@
 """MPI_Send() implementation"""
+
 import ctypes
 
 import numba

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/size.py

@@ -1,4 +1,5 @@
 """MPI_Comm_size() implementation"""
+
 import ctypes
 
 import numba

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/api/wtime.py

@@ -1,4 +1,5 @@
 """MPI_Wtime() implementation"""
+
 import ctypes
 
 import numba

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/common.py

@@ -1,4 +1,5 @@
 """variables used across API implementation"""
+
 import ctypes
 import os
 from ctypes.util import find_library

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi/utils.py

@@ -1,4 +1,5 @@
 """helper functions used across API implementation"""
+
 import numba
 import numpy as np
 from numba.core import cgutils, types

numba_mpi-0.41/numba_mpi.egg-info/PKG-INFO

@@ -0,0 +1,172 @@
+Metadata-Version: 2.1
+Name: numba-mpi
+Version: 0.41
+Summary: Numba @njittable MPI wrappers tested on Linux, macOS and Windows
+Home-page: https://github.com/numba-mpi/numba-mpi
+Author: https://github.com/numba-mpi/numba-mpi/graphs/contributors
+License: GPL v3
+Project-URL: Tracker, https://github.com/numba-mpi/numba-mpi/issues
+Project-URL: Documentation, https://numba-mpi.github.io/numba-mpi
+Project-URL: Source, https://github.com/numba-mpi/numba-mpi
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numba
+Requires-Dist: numpy
+Requires-Dist: mpi4py
+Requires-Dist: psutil
+Provides-Extra: tests
+Requires-Dist: pytest<8.0.0; extra == "tests"
+
+# <img src="https://raw.githubusercontent.com/numba-mpi/numba-mpi/main/.github/numba_mpi_logo.svg" style="height:50pt" alt="numba-mpi logo"> numba-mpi
+
+[![Python 3](https://img.shields.io/static/v1?label=Python&logo=Python&color=3776AB&message=3)](https://www.python.org/)
+[![LLVM](https://img.shields.io/static/v1?label=LLVM&logo=LLVM&color=gold&message=Numba)](https://numba.pydata.org)
+[![Linux OK](https://img.shields.io/static/v1?label=Linux&logo=Linux&color=yellow&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Linux)
+[![macOS OK](https://img.shields.io/static/v1?label=macOS&logo=Apple&color=silver&message=%E2%9C%93)](https://en.wikipedia.org/wiki/macOS)
+[![Windows OK](https://img.shields.io/static/v1?label=Windows&logo=Windows&color=white&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Windows)
+[![Github Actions Status](https://github.com/numba-mpi/numba-mpi/workflows/tests+pypi/badge.svg?branch=main)](https://github.com/numba-mpi/numba-mpi/actions/workflows/tests+pypi.yml)
+[![Maintenance](https://img.shields.io/badge/Maintained%3F-yes-green.svg)](https://GitHub.com/numba-mpi/numba-mpi/graphs/commit-activity)
+[![License: GPL v3](https://img.shields.io/badge/License-GPL%20v3-blue.svg)](https://www.gnu.org/licenses/gpl-3.0.html)
+[![PyPI version](https://badge.fury.io/py/numba-mpi.svg)](https://pypi.org/project/numba-mpi)
+[![Anaconda-Server Badge](https://anaconda.org/conda-forge/numba-mpi/badges/version.svg)](https://anaconda.org/conda-forge/numba-mpi)
+[![AUR package](https://repology.org/badge/version-for-repo/aur/python:numba-mpi.svg)](https://aur.archlinux.org/packages/python-numba-mpi)
+[![DOI](https://zenodo.org/badge/316911228.svg)](https://zenodo.org/badge/latestdoi/316911228)
+
+### Overview
+numba-mpi provides Python wrappers to the C MPI API callable from within [Numba JIT-compiled code](https://numba.readthedocs.io/en/stable/user/jit.html) (@njit mode).
+
+Support is provided for a subset of MPI routines covering: `size`/`rank`, `send`/`recv`, `allreduce`, `bcast`, `scatter`/`gather` & `allgather`, `barrier`, `wtime`
+and basic asynchronous communication with `isend`/`irecv` (only for contiguous arrays); for request handling including `wait`/`waitall`/`waitany` and `test`/`testall`/`testany`.
+
+The API uses NumPy and supports both numeric and character datatypes (e.g., `broadcast`). 
+Auto-generated docstring-based API docs are published on the web: https://numba-mpi.github.io/numba-mpi
+
+Packages can be obtained from 
+  [PyPI](https://pypi.org/project/numba-mpi), 
+  [Conda Forge](https://anaconda.org/conda-forge/numba-mpi), 
+  [Arch Linux](https://aur.archlinux.org/packages/python-numba-mpi)
+  or by invoking `pip install git+https://github.com/numba-mpi/numba-mpi.git`.
+
+numba-mpi is a pure-Python package.
+The codebase includes a test suite used through the GitHub Actions workflows ([thanks to mpi4py's setup-mpi](https://github.com/mpi4py/setup-mpi)!)
+for automated testing on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/) 
+& [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)), 
+macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and 
+Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi)).
+
+Features that are not implemented yet include (help welcome!):
+- support for non-default communicators
+- support for `MPI_IN_PLACE` in `[all]gather`/`scatter` and `allreduce`
+- support for `MPI_Type_create_struct` (Numpy structured arrays) 
+- ...
+
+### Hello world send/recv example:
+```python
+import numba, numba_mpi, numpy
+
+@numba.njit()
+def hello():
+    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()
+```
+
+### Example comparing numba-mpi vs. mpi4py performance:
+
+The example below compares Numba + mpi4py vs. Numba + numba-mpi performance.
+The sample code estimates $\pi$ by integration of $4/(1+x^2)$ between 0 and 1
+dividing the workload into `n_intervals` handled by separate MPI processes 
+and then obtaining a sum using `allreduce`.
+The computation is carried out in a JIT-compiled function and is repeated
+`N_TIMES`, the repetitions and the MPI-handled reduction are done outside or 
+inside of the JIT-compiled block for mpi4py and numba-mpi, respectively.
+Timing is repeated `N_REPEAT` times and the minimum time is reported.
+The generated plot shown below depicts the speedup obtained by replacing mpi4py
+with numba_mpi as a function of `n_intervals` - the more often communication
+is needed (smaller `n_intervals`), the larger the expected speedup.
+
+```python
+import timeit, mpi4py, numba, numpy as np, numba_mpi
+
+N_TIMES = 10000
+N_REPEAT = 10
+RTOL = 1e-3
+
+@numba.njit
+def get_pi_part(out, n_intervals, rank, size):
+    h = 1 / n_intervals
+    partial_sum = 0.0
+    for i in range(rank + 1, n_intervals, size):
+        x = h * (i - 0.5)
+        partial_sum += 4 / (1 + x**2)
+    out[0] = h * partial_sum
+
+@numba.njit
+def pi_numba_mpi(n_intervals):
+    pi = np.array([0.])
+    part = np.empty_like(pi)
+    for _ in range(N_TIMES):
+        get_pi_part(part, n_intervals, numba_mpi.rank(), numba_mpi.size())
+        numba_mpi.allreduce(part, pi, numba_mpi.Operator.SUM)
+        assert abs(pi[0] - np.pi) / np.pi < RTOL
+
+def pi_mpi4py(n_intervals):
+    pi = np.array([0.])
+    part = np.empty_like(pi)
+    for _ in range(N_TIMES):
+        get_pi_part(part, n_intervals, mpi4py.MPI.COMM_WORLD.rank, mpi4py.MPI.COMM_WORLD.size)
+        mpi4py.MPI.COMM_WORLD.Allreduce(part, (pi, mpi4py.MPI.DOUBLE), op=mpi4py.MPI.SUM)
+        assert abs(pi[0] - np.pi) / np.pi < RTOL
+
+plot_x = [1000 * k for k in range(1, 11)]
+plot_y = {'numba_mpi': [], 'mpi4py': []}
+for n_intervals in plot_x:
+    for impl in plot_y:
+        plot_y[impl].append(min(timeit.repeat(
+            f"pi_{impl}({n_intervals})",
+            globals=locals(),
+            number=1,
+            repeat=N_REPEAT
+        )))
+
+if numba_mpi.rank() == 0:
+    from matplotlib import pyplot
+    pyplot.figure(figsize=(8.3, 3.5), tight_layout=True)
+    pyplot.plot(plot_x, np.array(plot_y['mpi4py'])/np.array(plot_y['numba_mpi']), marker='o')
+    pyplot.xlabel('n_intervals (workload in between communication)')
+    pyplot.ylabel('wall time ratio (mpi4py / numba_mpi)')
+    pyplot.title(f'mpiexec -np {numba_mpi.size()}')
+    pyplot.grid()
+    pyplot.savefig('readme_plot.png')
+```
+
+![plot](https://github.com/numba-mpi/numba-mpi/releases/download/tip/readme_plot.png)
+
+
+### MPI resources on the web:
+
+- MPI standard and general information:
+    - https://www.mpi-forum.org/docs
+    - https://en.wikipedia.org/wiki/Message_Passing_Interface
+- MPI implementations:
+    - OpenMPI: https://www.open-mpi.org
+    - MPICH: https://www.mpich.org
+    - MS MPI: https://learn.microsoft.com/en-us/message-passing-interface
+    - Intel MPI: https://intel.com/content/www/us/en/developer/tools/oneapi/mpi-library-documentation.html
+- MPI bindings:
+    - Python: https://mpi4py.readthedocs.io
+    - Julia: https://juliaparallel.org/MPI.jl
+    - Rust: https://docs.rs/mpi
+    - C++: https://boost.org/doc/html/mpi.html
+    - R: https://cran.r-project.org/web/packages/Rmpi
+
+### Acknowledgements:
+
+Development of numba-mpi has been supported by the [Polish National Science Centre](https://ncn.gov.pl/en) (grant no. 2020/39/D/ST10/01220).
+

{numba-mpi-0.39 → numba_mpi-0.41}/numba_mpi.egg-info/requires.txt

@@ -2,3 +2,6 @@ numba
 numpy
 mpi4py
 psutil
+
+[tests]
+pytest<8.0.0

{numba-mpi-0.39 → numba_mpi-0.41}/setup.py

@@ -1,4 +1,5 @@
 """ the magick behind ``pip install ...`` """
+
 from setuptools import find_packages, setup
 
 
@@ -21,6 +22,7 @@ setup(
     license="GPL v3",
     description="Numba @njittable MPI wrappers tested on Linux, macOS and Windows",
     install_requires=("numba", "numpy", "mpi4py", "psutil"),
+    extras_require={"tests": ("pytest<8.0.0")},  # TODO #122
     long_description=get_long_description(),
     long_description_content_type="text/markdown",
     packages=find_packages(include=["numba_mpi", "numba_mpi.*"]),

numba-mpi-0.39/PKG-INFO

@@ -1,63 +0,0 @@
-Metadata-Version: 2.1
-Name: numba-mpi
-Version: 0.39
-Summary: Numba @njittable MPI wrappers tested on Linux, macOS and Windows
-Home-page: https://github.com/numba-mpi/numba-mpi
-Author: https://github.com/numba-mpi/numba-mpi/graphs/contributors
-License: GPL v3
-Project-URL: Tracker, https://github.com/numba-mpi/numba-mpi/issues
-Project-URL: Documentation, https://numba-mpi.github.io/numba-mpi
-Project-URL: Source, https://github.com/numba-mpi/numba-mpi
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: numba
-Requires-Dist: numpy
-Requires-Dist: mpi4py
-Requires-Dist: psutil
-
-# <img src="https://raw.githubusercontent.com/numba-mpi/numba-mpi/main/.github/numba_mpi_logo.svg" style="height:50pt" alt="numba-mpi logo"> numba-mpi
-
-[![Python 3](https://img.shields.io/static/v1?label=Python&logo=Python&color=3776AB&message=3)](https://www.python.org/)
-[![LLVM](https://img.shields.io/static/v1?label=LLVM&logo=LLVM&color=gold&message=Numba)](https://numba.pydata.org)
-[![Linux OK](https://img.shields.io/static/v1?label=Linux&logo=Linux&color=yellow&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Linux)
-[![macOS OK](https://img.shields.io/static/v1?label=macOS&logo=Apple&color=silver&message=%E2%9C%93)](https://en.wikipedia.org/wiki/macOS)
-[![Windows OK](https://img.shields.io/static/v1?label=Windows&logo=Windows&color=white&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Windows)
-[![Github Actions Status](https://github.com/numba-mpi/numba-mpi/workflows/tests+pypi/badge.svg?branch=main)](https://github.com/numba-mpi/numba-mpi/actions/workflows/tests+pypi.yml)
-[![Maintenance](https://img.shields.io/badge/Maintained%3F-yes-green.svg)](https://GitHub.com/numba-mpi/numba-mpi/graphs/commit-activity)
-[![License: GPL v3](https://img.shields.io/badge/License-GPL%20v3-blue.svg)](https://www.gnu.org/licenses/gpl-3.0.html)
-[![PyPI version](https://badge.fury.io/py/numba-mpi.svg)](https://pypi.org/project/numba-mpi)
-[![Anaconda-Server Badge](https://anaconda.org/conda-forge/numba-mpi/badges/version.svg)](https://anaconda.org/conda-forge/numba-mpi)
-[![DOI](https://zenodo.org/badge/316911228.svg)](https://zenodo.org/badge/latestdoi/316911228)
-
-#### Numba @njittable MPI wrappers
-- covering: `size`/`rank`, `send`/`recv`, `allreduce`, `bcast`, `scatter`/`gather` & `allgather`, `barrier` and `wtime`
-- basic asynchronous communication with `isend`/`irecv` (only for contiguous arrays); for request handling including `wait`/`waitall`/`waitany` and `test`/`testall`/`testany`
-- not yet implemented: support for non-default communicators, ...
-- API based on NumPy and supporting numeric and character datatypes 
-- auto-generated docstring-based API docs on the web: https://numba-mpi.github.io/numba-mpi
-- pure-Python implementation with packages available on [PyPI](https://pypi.org/project/numba-mpi) and [Conda Forge](https://anaconda.org/conda-forge/numba-mpi)
-- CI-tested on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/) & [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)), macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi))
-
-Hello world example:
-```python
-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://mpi4py.readthedocs.io/en/stable/

numba-mpi-0.39/README.md

@@ -1,46 +0,0 @@
-# <img src="https://raw.githubusercontent.com/numba-mpi/numba-mpi/main/.github/numba_mpi_logo.svg" style="height:50pt" alt="numba-mpi logo"> numba-mpi
-
-[![Python 3](https://img.shields.io/static/v1?label=Python&logo=Python&color=3776AB&message=3)](https://www.python.org/)
-[![LLVM](https://img.shields.io/static/v1?label=LLVM&logo=LLVM&color=gold&message=Numba)](https://numba.pydata.org)
-[![Linux OK](https://img.shields.io/static/v1?label=Linux&logo=Linux&color=yellow&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Linux)
-[![macOS OK](https://img.shields.io/static/v1?label=macOS&logo=Apple&color=silver&message=%E2%9C%93)](https://en.wikipedia.org/wiki/macOS)
-[![Windows OK](https://img.shields.io/static/v1?label=Windows&logo=Windows&color=white&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Windows)
-[![Github Actions Status](https://github.com/numba-mpi/numba-mpi/workflows/tests+pypi/badge.svg?branch=main)](https://github.com/numba-mpi/numba-mpi/actions/workflows/tests+pypi.yml)
-[![Maintenance](https://img.shields.io/badge/Maintained%3F-yes-green.svg)](https://GitHub.com/numba-mpi/numba-mpi/graphs/commit-activity)
-[![License: GPL v3](https://img.shields.io/badge/License-GPL%20v3-blue.svg)](https://www.gnu.org/licenses/gpl-3.0.html)
-[![PyPI version](https://badge.fury.io/py/numba-mpi.svg)](https://pypi.org/project/numba-mpi)
-[![Anaconda-Server Badge](https://anaconda.org/conda-forge/numba-mpi/badges/version.svg)](https://anaconda.org/conda-forge/numba-mpi)
-[![DOI](https://zenodo.org/badge/316911228.svg)](https://zenodo.org/badge/latestdoi/316911228)
-
-#### Numba @njittable MPI wrappers
-- covering: `size`/`rank`, `send`/`recv`, `allreduce`, `bcast`, `scatter`/`gather` & `allgather`, `barrier` and `wtime`
-- basic asynchronous communication with `isend`/`irecv` (only for contiguous arrays); for request handling including `wait`/`waitall`/`waitany` and `test`/`testall`/`testany`
-- not yet implemented: support for non-default communicators, ...
-- API based on NumPy and supporting numeric and character datatypes 
-- auto-generated docstring-based API docs on the web: https://numba-mpi.github.io/numba-mpi
-- pure-Python implementation with packages available on [PyPI](https://pypi.org/project/numba-mpi) and [Conda Forge](https://anaconda.org/conda-forge/numba-mpi)
-- CI-tested on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/) & [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)), macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi))
-
-Hello world example:
-```python
-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://mpi4py.readthedocs.io/en/stable/

numba-mpi-0.39/numba_mpi.egg-info/PKG-INFO

@@ -1,63 +0,0 @@
-Metadata-Version: 2.1
-Name: numba-mpi
-Version: 0.39
-Summary: Numba @njittable MPI wrappers tested on Linux, macOS and Windows
-Home-page: https://github.com/numba-mpi/numba-mpi
-Author: https://github.com/numba-mpi/numba-mpi/graphs/contributors
-License: GPL v3
-Project-URL: Tracker, https://github.com/numba-mpi/numba-mpi/issues
-Project-URL: Documentation, https://numba-mpi.github.io/numba-mpi
-Project-URL: Source, https://github.com/numba-mpi/numba-mpi
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: numba
-Requires-Dist: numpy
-Requires-Dist: mpi4py
-Requires-Dist: psutil
-
-# <img src="https://raw.githubusercontent.com/numba-mpi/numba-mpi/main/.github/numba_mpi_logo.svg" style="height:50pt" alt="numba-mpi logo"> numba-mpi
-
-[![Python 3](https://img.shields.io/static/v1?label=Python&logo=Python&color=3776AB&message=3)](https://www.python.org/)
-[![LLVM](https://img.shields.io/static/v1?label=LLVM&logo=LLVM&color=gold&message=Numba)](https://numba.pydata.org)
-[![Linux OK](https://img.shields.io/static/v1?label=Linux&logo=Linux&color=yellow&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Linux)
-[![macOS OK](https://img.shields.io/static/v1?label=macOS&logo=Apple&color=silver&message=%E2%9C%93)](https://en.wikipedia.org/wiki/macOS)
-[![Windows OK](https://img.shields.io/static/v1?label=Windows&logo=Windows&color=white&message=%E2%9C%93)](https://en.wikipedia.org/wiki/Windows)
-[![Github Actions Status](https://github.com/numba-mpi/numba-mpi/workflows/tests+pypi/badge.svg?branch=main)](https://github.com/numba-mpi/numba-mpi/actions/workflows/tests+pypi.yml)
-[![Maintenance](https://img.shields.io/badge/Maintained%3F-yes-green.svg)](https://GitHub.com/numba-mpi/numba-mpi/graphs/commit-activity)
-[![License: GPL v3](https://img.shields.io/badge/License-GPL%20v3-blue.svg)](https://www.gnu.org/licenses/gpl-3.0.html)
-[![PyPI version](https://badge.fury.io/py/numba-mpi.svg)](https://pypi.org/project/numba-mpi)
-[![Anaconda-Server Badge](https://anaconda.org/conda-forge/numba-mpi/badges/version.svg)](https://anaconda.org/conda-forge/numba-mpi)
-[![DOI](https://zenodo.org/badge/316911228.svg)](https://zenodo.org/badge/latestdoi/316911228)
-
-#### Numba @njittable MPI wrappers
-- covering: `size`/`rank`, `send`/`recv`, `allreduce`, `bcast`, `scatter`/`gather` & `allgather`, `barrier` and `wtime`
-- basic asynchronous communication with `isend`/`irecv` (only for contiguous arrays); for request handling including `wait`/`waitall`/`waitany` and `test`/`testall`/`testany`
-- not yet implemented: support for non-default communicators, ...
-- API based on NumPy and supporting numeric and character datatypes 
-- auto-generated docstring-based API docs on the web: https://numba-mpi.github.io/numba-mpi
-- pure-Python implementation with packages available on [PyPI](https://pypi.org/project/numba-mpi) and [Conda Forge](https://anaconda.org/conda-forge/numba-mpi)
-- CI-tested on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/) & [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)), macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi))
-
-Hello world example:
-```python
-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://mpi4py.readthedocs.io/en/stable/