numba-mpi 0.43__py3-none-any.whl → 1.0.1__py3-none-any.whl

numba_mpi/__init__.py

@@ -20,6 +20,8 @@ from .api.size import size
 from .api.wtime import wtime
 from .common import RequestType
 
+SUCCESS = 0
+
 try:
     __version__ = version(__name__)
 except PackageNotFoundError:

numba_mpi/api/initialized.py

@@ -17,5 +17,6 @@ def initialized():
     """wrapper for MPI_Initialized()"""
     flag = np.empty((1,), dtype=np.intc)
     status = _MPI_Initialized(flag.ctypes.data)
-    assert status == 0
+    if status != 0:
+        return False
     return bool(flag[0])

numba_mpi/api/rank.py

@@ -15,8 +15,9 @@ _MPI_Comm_rank.argtypes = [_MpiComm, ctypes.c_void_p]
 
 @numba.njit()
 def rank():
-    """wrapper for MPI_Comm_rank()"""
+    """wrapper for MPI_Comm_rank(), in case of failure returns 0"""
     value = np.empty(1, dtype=np.intc)
     status = _MPI_Comm_rank(_mpi_addr(_MPI_Comm_World_ptr), value.ctypes.data)
-    assert status == 0
+    if status != 0:
+        value[0] = 0
     return value[0]

numba_mpi/api/size.py

@@ -15,8 +15,9 @@ _MPI_Comm_size.argtypes = [_MpiComm, ctypes.c_void_p]
 
 @numba.njit()
 def size():
-    """wrapper for MPI_Comm_size()"""
+    """wrapper for MPI_Comm_size(), in case of failure returns 0"""
     value = np.empty(1, dtype=np.intc)
     status = _MPI_Comm_size(_mpi_addr(_MPI_Comm_World_ptr), value.ctypes.data)
-    assert status == 0
+    if status != 0:
+        value[0] = 0
     return value[0]

{numba_mpi-0.43.dist-info → numba_mpi-1.0.1.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: numba-mpi
-Version: 0.43
-Summary: Numba @njittable MPI wrappers tested on Linux, macOS and Windows
+Version: 1.0.1
+Summary: Numba @jittable 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
@@ -17,8 +17,9 @@ Requires-Dist: mpi4py
 Requires-Dist: psutil
 Provides-Extra: tests
 Requires-Dist: pytest <8.0.0 ; extra == 'tests'
+Requires-Dist: py-pde ; 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
+# <img src="https://raw.githubusercontent.com/numba-mpi/numba-mpi/main/.github/numba_mpi_logo.png" width=128 height=142 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)
@@ -34,7 +35,7 @@ Requires-Dist: pytest <8.0.0 ; extra == 'tests'
 [![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).
+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) (@jit mode). For an outline of the project, rationale, architecture, and features, refer to: [numba-mpi arXiv e-print](https://doi.org/10.48550/arXiv.2407.13712) (please cite if numba-mpi is used in your research).
 
 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`.
@@ -65,7 +66,7 @@ Features that are not implemented yet include (help welcome!):
 ```python
 import numba, numba_mpi, numpy
 
-@numba.njit()
+@numba.jit()
 def hello():
     src = numpy.array([1., 2., 3., 4., 5.])
     dst_tst = numpy.empty_like(src)
@@ -168,6 +169,7 @@ if numba_mpi.rank() == 0:
     - Intel MPI: https://intel.com/content/www/us/en/developer/tools/oneapi/mpi-library-documentation.html
 - MPI bindings:
     - Python: https://mpi4py.readthedocs.io
+    - Python/JAX: https://mpi4jax.readthedocs.io
     - Julia: https://juliaparallel.org/MPI.jl
     - Rust: https://docs.rs/mpi
     - C++: https://boost.org/doc/html/mpi.html
@@ -175,5 +177,11 @@ if numba_mpi.rank() == 0:
 
 ### 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).
+We thank [all contributors](https://github.com/numba-mpi/numba-mpi/graphs/contributors) and users who reported feedback to the project 
+  through [GitHub issues](https://github.com/numba-mpi/numba-mpi/issues).
+
+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),
+  the [Max Planck Society](https://www.mpg.de/en) and the [European Union](https://erc.europa.eu/) (ERC, EmulSim, 101044662). 
+We further acknowledge Poland’s high-performance computing infrastructure [PLGrid](https://plgrid.pl) (HPC Centers: [ACK Cyfronet AGH](https://www.cyfronet.pl/en)) 
+  for providing computer facilities and support within computational grant no. PLG/2023/016369.
 

{numba_mpi-0.43.dist-info → numba_mpi-1.0.1.dist-info}/RECORD

@@ -1,23 +1,23 @@
-numba_mpi/__init__.py,sha256=VkDzjRJcaS6j739oHdCqhdUcZLBlMFxfcn58zDjevoQ,741
+numba_mpi/__init__.py,sha256=_DsPxgrR80KiJTLqzZRNMVsK_TUJt7EfNFy_MWvBOWk,754
 numba_mpi/common.py,sha256=2JJoUrd3Qa6GIFk6Zlt2NudS7ZurPxpVwBLRGSkCg5E,2266
 numba_mpi/utils.py,sha256=gfGFuzmGgs4FnBqzPI91ftAq4UHgXb_HFkvxrVWkcIo,1866
 numba_mpi/api/__init__.py,sha256=Zj5df4lWeGpxAXV8jKGFnmtLBQ50HwNU8dPf-os06X8,51
 numba_mpi/api/allreduce.py,sha256=szS7YzrQ5a90LlKDiefyxVEiAXnsHbni5g2M1of0TmE,3261
 numba_mpi/api/barrier.py,sha256=9VSJPBC4V0H-xo47uzlT8Hp4xmQhTNLxg5bAcX3Y03g,461
 numba_mpi/api/bcast.py,sha256=8SsYFj9qRjx4l3Q9367JMZd469izlf4if1qusuYILqU,1843
-numba_mpi/api/initialized.py,sha256=fFmhOGl2GkwS9UvPetXy4YrgE4xhMVsyx7Ac8QP0530,479
+numba_mpi/api/initialized.py,sha256=oKXpZzHeips0VU1U9wEF_578kOrfKb_IEXxD_aQ2c2E,497
 numba_mpi/api/irecv.py,sha256=r4JvE7JJPN_hFpS79-idYL3dtp8tR0y0VoIRuHJ29lM,1120
 numba_mpi/api/isend.py,sha256=2mpP4FhMk0GrikjDluKwRnpVywdLj9RD4HVVEMSj9A8,1080
 numba_mpi/api/operator.py,sha256=3VTPZAdOP05bxdqt3lA0hRDICM-iaBMa4m-krEdO91s,342
-numba_mpi/api/rank.py,sha256=pqayxw-5QDJ7VJ3gKrvuu1G0sBlYEZt1juhnaDi_JD8,549
+numba_mpi/api/rank.py,sha256=1xZvHUclsK20aMtK07JzXYxW5F4Er8HZgOmcf495sjo,597
 numba_mpi/api/recv.py,sha256=YsYK-q7PNfi3zt0ftVddM363VsnJ4XFfmgMq8aeCr-o,1260
 numba_mpi/api/requests.py,sha256=5EhgFyeQCGP8YclSPwxP95c2AhBo19CLlShK0TxCR2U,9114
 numba_mpi/api/scatter_gather.py,sha256=goZn4BxMKakWQHjfXIOdjzK3DJ-lTeaiQQwgnyQeZ_s,2410
 numba_mpi/api/send.py,sha256=jn1hPw0YHBHOaeJop_ZbjaBChaqgfw3nM1xGhW9sabI,909
-numba_mpi/api/size.py,sha256=fYLeUrygvz_XcxIDsLiZlMtS-aiWfp58Zi7aIOAgaj8,549
+numba_mpi/api/size.py,sha256=-RX-FtcIH4qDxCoGOhZjjgEWXpytt79vsH0YX9dtZuY,597
 numba_mpi/api/wtime.py,sha256=qrTqlefW7K7hqnAQKkGYm8kgdiRGuSAGiHmPcTrhLzE,279
-numba_mpi-0.43.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-numba_mpi-0.43.dist-info/METADATA,sha256=oMga9o3BuVvyY-f-9tCWPqkYjj06J7cIbhLZHGZgW4g,8995
-numba_mpi-0.43.dist-info/WHEEL,sha256=cpQTJ5IWu9CdaPViMhC9YzF8gZuS5-vlfoFihTBC86A,91
-numba_mpi-0.43.dist-info/top_level.txt,sha256=yb_ktLmrfuhOZS0rjS81FFNC-gK_4c19WbLG2ViP73g,10
-numba_mpi-0.43.dist-info/RECORD,,
+numba_mpi-1.0.1.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+numba_mpi-1.0.1.dist-info/METADATA,sha256=Fp5jqI53wUJzoY-KreJqYy_tiXzuCyJNGQyuFEkYzC0,9887
+numba_mpi-1.0.1.dist-info/WHEEL,sha256=uCRv0ZEik_232NlR4YDw4Pv3Ajt5bKvMH13NUU7hFuI,91
+numba_mpi-1.0.1.dist-info/top_level.txt,sha256=yb_ktLmrfuhOZS0rjS81FFNC-gK_4c19WbLG2ViP73g,10
+numba_mpi-1.0.1.dist-info/RECORD,,

{numba_mpi-0.43.dist-info → numba_mpi-1.0.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (70.1.0)
+Generator: setuptools (74.1.1)
 Root-Is-Purelib: true
 Tag: py3-none-any