numba-mpi 1.1.4__tar.gz → 1.1.5__tar.gz

{numba_mpi-1.1.4 → numba_mpi-1.1.5}/.zenodo.json

@@ -3,7 +3,7 @@
   "description": "Numba @jittable wrappers for MPI C API tested on Linux, macOS and Windows",
   "creators": [
     {
-      "affiliation": "Jagiellonian University, Kraków, Poland",
+      "affiliation": "Jagiellonian University, Kraków, Poland (-2023); AGH University of Krakow, Poland (2023-)",
       "name": "Arabas, Sylwester",
       "orcid": "0000-0003-2361-0082"
     },
@@ -21,6 +21,10 @@
       "name": "Manna, Maciej",
       "orcid": "0000-0002-7482-4913"
     },
+    {
+      "affiliation": "AGH University of Krakow, Poland",
+      "name": "Michał Wroński"
+    },
     {
       "affiliation": "Max Planck Institute for Dynamics and Self-Organization, Göttingen, Germany",
       "name": "Zwicker, David",
@@ -33,4 +37,4 @@
   ],
   "license": "GPL-3.0-only",
   "upload_type": "software"
-}
+}

{numba_mpi-1.1.4/numba_mpi.egg-info → numba_mpi-1.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numba-mpi
-Version: 1.1.4
+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

{numba_mpi-1.1.4 → numba_mpi-1.1.5}/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-1.1.5/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-1.1.4 → numba_mpi-1.1.5/numba_mpi.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: numba-mpi
-Version: 1.1.4
+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

{numba_mpi-1.1.4 → numba_mpi-1.1.5}/numba_mpi.egg-info/SOURCES.txt

@@ -28,6 +28,7 @@ numba_mpi/api/initialized.py
 numba_mpi/api/irecv.py
 numba_mpi/api/isend.py
 numba_mpi/api/operator.py
+numba_mpi/api/query_thread.py
 numba_mpi/api/rank.py
 numba_mpi/api/recv.py
 numba_mpi/api/reduce.py
@@ -46,6 +47,7 @@ tests/api/test_barrier.py
 tests/api/test_bcast.py
 tests/api/test_init.py
 tests/api/test_isend_irecv.py
+tests/api/test_query_thread.py
 tests/api/test_rank.py
 tests/api/test_reduce.py
 tests/api/test_scatter_gather.py

numba_mpi-1.1.5/tests/api/test_query_thread.py

@@ -0,0 +1,8 @@
+# pylint: disable=missing-function-docstring,missing-module-docstring
+from mpi4py import MPI
+
+import numba_mpi as mpi
+
+
+def test_query_thread():
+    assert MPI.Query_thread() == mpi.query_thread()