numba-mpi 1.1.3__py3-none-any.whl → 1.1.5__py3-none-any.whl

numba_mpi/__init__.py

@@ -11,6 +11,7 @@ from .api.initialized import initialized
 from .api.irecv import irecv
 from .api.isend import isend
 from .api.operator import Operator
+from .api.query_thread import query_thread
 from .api.rank import rank
 from .api.recv import recv
 from .api.reduce import reduce

numba_mpi/api/query_thread.py

@@ -0,0 +1,20 @@
+"""file contains MPI_Query_thread() implementation"""
+
+import ctypes
+
+import numba
+import numpy as np
+
+from numba_mpi.common import libmpi
+
+_MPI_Query_thread = libmpi.MPI_Query_thread
+_MPI_Query_thread.restype = ctypes.c_int
+_MPI_Query_thread.argtypes = [ctypes.c_void_p]
+
+
+@numba.njit()
+def query_thread():
+    """wrapper for MPI_Query_thread()"""
+    provided = np.empty(1, dtype=np.intc)
+    _ = _MPI_Query_thread(provided.ctypes.data)
+    return provided[0]

numba_mpi/common.py

@@ -2,6 +2,7 @@
 
 import ctypes
 import os
+import sys
 from ctypes.util import find_library
 from pathlib import Path
 
@@ -72,6 +73,12 @@ else:
             break
 
 if LIB is None:
+    if sys.platform == "darwin":
+        raise RuntimeError(
+            """MPI library not found, if MPI was installed with Homebrew, export the following:
+            DYLD_FALLBACK_LIBRARY_PATH="/opt/homebrew/lib:/usr/lib:$DYLD_FALLBACK_LIBRARY_PATH
+            """
+        )
     raise RuntimeError("no MPI library found")
 
 libmpi = ctypes.CDLL(LIB)

{numba_mpi-1.1.3.dist-info → numba_mpi-1.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: numba-mpi
-Version: 1.1.3
+Version: 1.1.5
 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
@@ -23,6 +23,7 @@ Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
 Dynamic: license
+Dynamic: license-file
 Dynamic: project-url
 Dynamic: provides-extra
 Dynamic: requires-dist

{numba_mpi-1.1.3.dist-info → numba_mpi-1.1.5.dist-info}/RECORD

@@ -1,5 +1,5 @@
-numba_mpi/__init__.py,sha256=EsiszqT6I7-Vrbgdk36_mGp5kHpAS5AYCkhjRya-5rg,785
-numba_mpi/common.py,sha256=2JJoUrd3Qa6GIFk6Zlt2NudS7ZurPxpVwBLRGSkCg5E,2266
+numba_mpi/__init__.py,sha256=UupJqiiT2YdyzF80ZOIR3Gc1HX6MY8EsDtFGYcSw_xA,828
+numba_mpi/common.py,sha256=XOnXiy2hDFIsNkhTUXuPY17Xz-a9xg6F-SxIayMrgqU,2555
 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
@@ -9,6 +9,7 @@ numba_mpi/api/initialized.py,sha256=oKXpZzHeips0VU1U9wEF_578kOrfKb_IEXxD_aQ2c2E,
 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/query_thread.py,sha256=2oVA2cYIeG6fcDJZGOetYrLoFyND--b-4qARGM_D5_E,462
 numba_mpi/api/rank.py,sha256=1xZvHUclsK20aMtK07JzXYxW5F4Er8HZgOmcf495sjo,597
 numba_mpi/api/recv.py,sha256=YsYK-q7PNfi3zt0ftVddM363VsnJ4XFfmgMq8aeCr-o,1260
 numba_mpi/api/reduce.py,sha256=ug8WctzXRuxD3IrvrXN4sAfWdjJ1vvMhwykTgF8nn1s,3233
@@ -17,8 +18,8 @@ numba_mpi/api/scatter_gather.py,sha256=goZn4BxMKakWQHjfXIOdjzK3DJ-lTeaiQQwgnyQeZ
 numba_mpi/api/send.py,sha256=jn1hPw0YHBHOaeJop_ZbjaBChaqgfw3nM1xGhW9sabI,909
 numba_mpi/api/size.py,sha256=-RX-FtcIH4qDxCoGOhZjjgEWXpytt79vsH0YX9dtZuY,597
 numba_mpi/api/wtime.py,sha256=qrTqlefW7K7hqnAQKkGYm8kgdiRGuSAGiHmPcTrhLzE,279
-numba_mpi-1.1.3.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-numba_mpi-1.1.3.dist-info/METADATA,sha256=cBFNRsbMKWsQVU0TA3q9VNGBQlxzCmTzHjP4W5Mo_TU,11215
-numba_mpi-1.1.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-numba_mpi-1.1.3.dist-info/top_level.txt,sha256=yb_ktLmrfuhOZS0rjS81FFNC-gK_4c19WbLG2ViP73g,10
-numba_mpi-1.1.3.dist-info/RECORD,,
+numba_mpi-1.1.5.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+numba_mpi-1.1.5.dist-info/METADATA,sha256=B0kTwYRyWmQw0iGnz1-rT1p3f10-RXBvfvH9aRQdlgA,11237
+numba_mpi-1.1.5.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+numba_mpi-1.1.5.dist-info/top_level.txt,sha256=yb_ktLmrfuhOZS0rjS81FFNC-gK_4c19WbLG2ViP73g,10
+numba_mpi-1.1.5.dist-info/RECORD,,

{numba_mpi-1.1.3.dist-info → numba_mpi-1.1.5.dist-info}/WHEEL

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