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

{numba-mpi-0.40 → 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.40 → numba_mpi-0.41}/.github/workflows/tests+pypi.yml

@@ -88,13 +88,13 @@ jobs:
     needs: [pylint, precommit, pdoc, zenodo_json]
     strategy:
       matrix:
-        platform: [ubuntu-latest, macos-latest, windows-latest]
+        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

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.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.40/PKG-INFO

@@ -1,66 +0,0 @@
-Metadata-Version: 2.1
-Name: numba-mpi
-Version: 0.40
-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)
-
-#### 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 at [PyPI](https://pypi.org/project/numba-mpi), [Conda Forge](https://anaconda.org/conda-forge/numba-mpi) and for [Arch Linux](https://aur.archlinux.org/packages/python-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.40/README.md

@@ -1,47 +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)
-[![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)
-
-#### 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 at [PyPI](https://pypi.org/project/numba-mpi), [Conda Forge](https://anaconda.org/conda-forge/numba-mpi) and for [Arch Linux](https://aur.archlinux.org/packages/python-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.40/numba_mpi.egg-info/PKG-INFO

@@ -1,66 +0,0 @@
-Metadata-Version: 2.1
-Name: numba-mpi
-Version: 0.40
-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)
-
-#### 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 at [PyPI](https://pypi.org/project/numba-mpi), [Conda Forge](https://anaconda.org/conda-forge/numba-mpi) and for [Arch Linux](https://aur.archlinux.org/packages/python-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/