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

{numba_mpi-1.1.3 → numba_mpi-1.1.5}/.github/workflows/tests+pypi.yml

@@ -93,7 +93,7 @@ jobs:
           - ubuntu-latest
           - windows-latest
           - macos-13          # arch: Intel
-          # - macos-14        # arch: arm64    # TODO: #164
+          - macos-15          # arch: arm64
         python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
         mpi: [ 'mpich', 'openmpi', 'msmpi', 'intelmpi']
         exclude:
@@ -103,10 +103,12 @@ jobs:
           mpi: intelmpi
         - platform: macos-13   # TODO: # 163
           mpi: openmpi
-        # - platform: macos-14
-        #   mpi: msmpi
-        # - platform: macos-14
-        #   mpi: intelmpi
+        - platform: macos-15
+          mpi: msmpi
+        - platform: macos-15
+          mpi: intelmpi
+        - platform: macos-15
+          mpi: openmpi
         - platform: ubuntu-latest
           mpi: msmpi
         - platform: ubuntu-latest # TODO: #162
@@ -132,6 +134,8 @@ jobs:
       - uses: mpi4py/setup-mpi@v1
         with:
           mpi: ${{ matrix.mpi }}
+      - if: ${{ matrix.platform == 'macos-15' }}
+        run: echo DYLD_FALLBACK_LIBRARY_PATH="/opt/homebrew/lib:/usr/lib:$DYLD_FALLBACK_LIBRARY_PATH" >> $GITHUB_ENV
       - if: matrix.mpi == 'intelmpi'
         run: |
           # checking if multiple MPI libs are OK
@@ -172,13 +176,22 @@ jobs:
           cd /tmp  # make sure local files are not picked up
           python -We -c "import numba_mpi"
 
-      - if: github.event_name == 'push' && github.ref == 'refs/heads/main'
+      - if: |-
+          ${{
+            github.repository == 'numba-mpi/numba-mpi' &&
+            github.event_name == 'push' &&
+            github.ref == 'refs/heads/main'
+          }}
         uses: pypa/gh-action-pypi-publish@release/v1.12
         with:
           attestations: false
           repository_url: https://test.pypi.org/legacy/
 
-      - if: startsWith(github.ref, 'refs/tags')
+      - if: |-
+          ${{
+            github.repository == 'numba-mpi/numba-mpi' &&
+            startsWith(github.ref, 'refs/tags')
+          }}
         uses: pypa/gh-action-pypi-publish@release/v1.12
         with:
           attestations: false

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

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

@@ -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 → 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.3 → numba_mpi-1.1.5}/tests/api/test_bcast.py

@@ -15,8 +15,8 @@ def jit_bcast(data, root):
 
 @pytest.mark.parametrize("bcast", (jit_bcast.py_func, jit_bcast))
 @pytest.mark.parametrize("data_type", data_types)
-def test_bcast_np_array(data_type, bcast):
-    root = 0
+@pytest.mark.parametrize("root", range(mpi.size()))
+def test_bcast_np_array(data_type, bcast, root):
     data = np.empty(5, data_type).astype(dtype=data_type)
     datatobcast = get_random_array(5, data_type).astype(dtype=data_type)
 
@@ -34,8 +34,8 @@ def test_bcast_np_array(data_type, bcast):
     "stringtobcast",
     ("test bcast", pytest.param("żółć", marks=pytest.mark.xfail(strict=True))),
 )
-def test_bcast_string(stringtobcast):
-    root = 0
+@pytest.mark.parametrize("root", range(mpi.size()))
+def test_bcast_string(stringtobcast, root):
     datatobcast = np.array(stringtobcast, "c")
     data = np.empty_like(datatobcast)
 

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()