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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (63) hide show
  1. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/.github/workflows/readme_snippets.yml +3 -3
  2. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/.github/workflows/tests+pypi.yml +28 -20
  3. {numba_mpi-1.1.1/numba_mpi.egg-info → numba_mpi-1.1.3}/PKG-INFO +40 -22
  4. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/README.md +28 -20
  5. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/requests.py +2 -2
  6. {numba_mpi-1.1.1 → numba_mpi-1.1.3/numba_mpi.egg-info}/PKG-INFO +40 -22
  7. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/setup.py +2 -2
  8. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/test_paper_listings.py +1 -1
  9. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/.github/numba_mpi_logo.png +0 -0
  10. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/.github/numba_mpi_logo.svg +0 -0
  11. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/.github/workflows/stale.yml +0 -0
  12. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/.gitignore +0 -0
  13. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/.pre-commit-config.yaml +0 -0
  14. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/.vscode/settings.json +0 -0
  15. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/.zenodo.json +0 -0
  16. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/CITATION.cff +0 -0
  17. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/CODE_OF_CONDUCT.md +0 -0
  18. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/LICENSE +0 -0
  19. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/__init__.py +0 -0
  20. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/__init__.py +0 -0
  21. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/allreduce.py +0 -0
  22. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/barrier.py +0 -0
  23. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/bcast.py +0 -0
  24. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/initialized.py +0 -0
  25. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/irecv.py +0 -0
  26. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/isend.py +0 -0
  27. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/operator.py +0 -0
  28. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/rank.py +0 -0
  29. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/recv.py +0 -0
  30. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/reduce.py +0 -0
  31. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/scatter_gather.py +0 -0
  32. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/send.py +0 -0
  33. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/size.py +0 -0
  34. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/api/wtime.py +0 -0
  35. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/common.py +0 -0
  36. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi/utils.py +0 -0
  37. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi.egg-info/SOURCES.txt +0 -0
  38. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi.egg-info/dependency_links.txt +0 -0
  39. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi.egg-info/requires.txt +0 -0
  40. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/numba_mpi.egg-info/top_level.txt +0 -0
  41. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/setup.cfg +0 -0
  42. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/__init__.py +0 -0
  43. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/api/test_allreduce.py +0 -0
  44. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/api/test_barrier.py +0 -0
  45. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/api/test_bcast.py +0 -0
  46. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/api/test_init.py +0 -0
  47. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/api/test_isend_irecv.py +0 -0
  48. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/api/test_rank.py +0 -0
  49. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/api/test_reduce.py +0 -0
  50. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/api/test_scatter_gather.py +0 -0
  51. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/api/test_send_recv.py +0 -0
  52. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/api/test_size.py +0 -0
  53. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/api/test_wtime.py +0 -0
  54. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/common.py +0 -0
  55. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/paper_listings/exchange.py +0 -0
  56. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/paper_listings/hello.py +0 -0
  57. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/paper_listings/mpi4py_with_error.py +0 -0
  58. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/paper_listings/numba_mpi.py +0 -0
  59. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/paper_listings/py-pde.py +0 -0
  60. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/paper_listings/test.py +0 -0
  61. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/paper_listings/timing.py +0 -0
  62. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/test_version.py +0 -0
  63. {numba_mpi-1.1.1 → numba_mpi-1.1.3}/tests/utils.py +0 -0
@@ -16,8 +16,8 @@ jobs:
16
16
  python:
17
17
  runs-on: ubuntu-latest
18
18
  steps:
19
- - uses: actions/checkout@v2
20
- - uses: actions/setup-python@v1
19
+ - uses: actions/checkout@v4
20
+ - uses: actions/setup-python@v5
21
21
  with:
22
22
  python-version: "3.10"
23
23
  - uses: mpi4py/setup-mpi@v1
@@ -28,7 +28,7 @@ jobs:
28
28
  - run: cat -n readme.py
29
29
  - run: mpiexec -n 4 python -We readme.py
30
30
  - run: rsvg-convert readme_plot.svg > readme_plot.png
31
- - uses: actions/upload-artifact@v3
31
+ - uses: actions/upload-artifact@v4
32
32
  with:
33
33
  path: readme_plot.*
34
34
  name: readme_plot
@@ -18,10 +18,10 @@ jobs:
18
18
  pylint:
19
19
  runs-on: ubuntu-latest
20
20
  steps:
21
- - uses: actions/checkout@v2
21
+ - uses: actions/checkout@v4
22
22
  - uses: mpi4py/setup-mpi@v1
23
23
  - name: Set up Python
24
- uses: actions/setup-python@v2
24
+ uses: actions/setup-python@v5
25
25
  with:
26
26
  python-version: "3.10"
27
27
  - name: Install dependencies
@@ -38,11 +38,11 @@ jobs:
38
38
  precommit:
39
39
  runs-on: ubuntu-latest
40
40
  steps:
41
- - uses: actions/checkout@v2
41
+ - uses: actions/checkout@v4
42
42
  - name: Set up Python
43
- uses: actions/setup-python@v1
43
+ uses: actions/setup-python@v5
44
44
  with:
45
- python-version: "3.10"
45
+ python-version: "3.10"
46
46
  - name: Linting
47
47
  run: |
48
48
  pip install pre-commit
@@ -55,10 +55,10 @@ jobs:
55
55
  permissions:
56
56
  contents: write
57
57
  steps:
58
- - uses: actions/checkout@v2
58
+ - uses: actions/checkout@v4
59
59
  with:
60
60
  persist-credentials: false
61
- - uses: actions/setup-python@v2
61
+ - uses: actions/setup-python@v5
62
62
  with:
63
63
  python-version: "3.10"
64
64
  - uses: mpi4py/setup-mpi@v1
@@ -78,9 +78,9 @@ jobs:
78
78
  zenodo_json:
79
79
  runs-on: ubuntu-latest
80
80
  steps:
81
- - uses: actions/checkout@v2
81
+ - uses: actions/checkout@v4
82
82
  - uses: notiz-dev/github-action-json-property@release
83
- with:
83
+ with:
84
84
  path: '.zenodo.json'
85
85
  prop_path: 'creators'
86
86
 
@@ -89,35 +89,44 @@ jobs:
89
89
  strategy:
90
90
  matrix:
91
91
  mpi-np: [2, 3]
92
- platform: [ubuntu-latest, macos-12, windows-latest]
93
- python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"]
92
+ platform:
93
+ - ubuntu-latest
94
+ - windows-latest
95
+ - macos-13 # arch: Intel
96
+ # - macos-14 # arch: arm64 # TODO: #164
97
+ python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
94
98
  mpi: [ 'mpich', 'openmpi', 'msmpi', 'intelmpi']
95
99
  exclude:
96
- - platform: macos-12
100
+ - platform: macos-13
97
101
  mpi: msmpi
98
- - platform: macos-12
102
+ - platform: macos-13
99
103
  mpi: intelmpi
104
+ - platform: macos-13 # TODO: # 163
105
+ mpi: openmpi
106
+ # - platform: macos-14
107
+ # mpi: msmpi
108
+ # - platform: macos-14
109
+ # mpi: intelmpi
100
110
  - platform: ubuntu-latest
101
111
  mpi: msmpi
112
+ - platform: ubuntu-latest # TODO: #162
113
+ mpi: mpich
102
114
  - platform: windows-latest
103
115
  mpi: mpich
104
116
  - platform: windows-latest
105
117
  mpi: openmpi
106
118
  - platform: windows-latest
107
119
  mpi: intelmpi
108
-
120
+
109
121
  # https://github.com/numba-mpi/numba-mpi/issues/69
110
- - platform: ubuntu-latest
111
- python-version: 3.8
112
- mpi: mpich
113
122
  - platform: ubuntu-latest
114
123
  python-version: 3.9
115
124
  mpi: mpich
116
125
  fail-fast: false
117
126
  runs-on: ${{ matrix.platform }}
118
127
  steps:
119
- - uses: actions/checkout@v2
120
- - uses: actions/setup-python@v1
128
+ - uses: actions/checkout@v4
129
+ - uses: actions/setup-python@v5
121
130
  with:
122
131
  python-version: ${{ matrix.python-version }}
123
132
  - uses: mpi4py/setup-mpi@v1
@@ -135,7 +144,6 @@ jobs:
135
144
 
136
145
  dist:
137
146
  runs-on: ubuntu-latest
138
- needs: [build]
139
147
  permissions:
140
148
  id-token: write
141
149
  steps:
@@ -1,6 +1,6 @@
1
- Metadata-Version: 2.1
1
+ Metadata-Version: 2.2
2
2
  Name: numba-mpi
3
- Version: 1.1.1
3
+ Version: 1.1.3
4
4
  Summary: Numba @jittable MPI wrappers tested on Linux, macOS and Windows
5
5
  Home-page: https://github.com/numba-mpi/numba-mpi
6
6
  Author: https://github.com/numba-mpi/numba-mpi/graphs/contributors
@@ -18,6 +18,16 @@ Requires-Dist: psutil
18
18
  Provides-Extra: tests
19
19
  Requires-Dist: pytest<8.0.0; extra == "tests"
20
20
  Requires-Dist: py-pde; extra == "tests"
21
+ Dynamic: author
22
+ Dynamic: description
23
+ Dynamic: description-content-type
24
+ Dynamic: home-page
25
+ Dynamic: license
26
+ Dynamic: project-url
27
+ Dynamic: provides-extra
28
+ Dynamic: requires-dist
29
+ Dynamic: requires-python
30
+ Dynamic: summary
21
31
 
22
32
  # <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
23
33
 
@@ -35,31 +45,32 @@ Requires-Dist: py-pde; extra == "tests"
35
45
  [![DOI](https://zenodo.org/badge/316911228.svg)](https://zenodo.org/badge/latestdoi/316911228)
36
46
 
37
47
  ### Overview
38
- 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).
48
+ 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 paper in SoftwareX (open access)](https://www.sciencedirect.com/science/article/pii/S235271102400267X) (please cite if numba-mpi is used in your research).
39
49
 
40
50
  Support is provided for a subset of MPI routines covering: `size`/`rank`, `send`/`recv`, `allreduce`, `reduce`, `bcast`, `scatter`/`gather` & `allgather`, `barrier`, `wtime`
41
51
  and basic asynchronous communication with `isend`/`irecv` (only for contiguous arrays); for request handling including `wait`/`waitall`/`waitany` and `test`/`testall`/`testany`.
42
52
 
43
- The API uses NumPy and supports both numeric and character datatypes (e.g., `broadcast`).
53
+ The API uses NumPy and supports both numeric and character datatypes (e.g., `broadcast`).
44
54
  Auto-generated docstring-based API docs are published on the web: https://numba-mpi.github.io/numba-mpi
45
55
 
46
- Packages can be obtained from
47
- [PyPI](https://pypi.org/project/numba-mpi),
48
- [Conda Forge](https://anaconda.org/conda-forge/numba-mpi),
56
+ Packages can be obtained from
57
+ [PyPI](https://pypi.org/project/numba-mpi),
58
+ [Conda Forge](https://anaconda.org/conda-forge/numba-mpi),
49
59
  [Arch Linux](https://aur.archlinux.org/packages/python-numba-mpi)
50
60
  or by invoking `pip install git+https://github.com/numba-mpi/numba-mpi.git`.
51
61
 
52
62
  numba-mpi is a pure-Python package.
53
63
  The codebase includes a test suite used through the GitHub Actions workflows ([thanks to mpi4py's setup-mpi](https://github.com/mpi4py/setup-mpi)!)
54
- for automated testing on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/)
55
- & [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)),
56
- macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and
57
- Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi)).
64
+ for automated testing on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/)
65
+ & [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)),
66
+ macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and
67
+ Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi)). Note, that some of those
68
+ combinations may not be fully supported yet - see [Known Issues](#known-issues) for more information.
58
69
 
59
70
  Features that are not implemented yet include (help welcome!):
60
71
  - support for non-default communicators
61
72
  - support for `MPI_IN_PLACE` in `[all]gather`/`scatter` and `allreduce`
62
- - support for `MPI_Type_create_struct` (Numpy structured arrays)
73
+ - support for `MPI_Type_create_struct` (Numpy structured arrays)
63
74
  - ...
64
75
 
65
76
  ### Hello world send/recv example:
@@ -82,21 +93,21 @@ hello()
82
93
  ### Example comparing numba-mpi vs. mpi4py performance:
83
94
 
84
95
  The example below compares `Numba`+`mpi4py` vs. `Numba`+`numba-mpi` performance.
85
- The sample code estimates $\pi$ by numerical integration of $\int_0^1 (4/(1+x^2))dx=\pi$
86
- dividing the workload into `n_intervals` handled by separate MPI processes
96
+ The sample code estimates $\pi$ by numerical integration of $\int_0^1 (4/(1+x^2))dx=\pi$
97
+ dividing the workload into `n_intervals` handled by separate MPI processes
87
98
  and then obtaining a sum using `allreduce` (see, e.g., analogous [Matlab docs example](https://www.mathworks.com/help/parallel-computing/numerical-estimation-of-pi-using-message-passing.html)).
88
99
  The computation is carried out in a JIT-compiled function `get_pi_part()` and is repeated
89
- `N_TIMES`. The repetitions and the MPI-handled reduction are done outside or
100
+ `N_TIMES`. The repetitions and the MPI-handled reduction are done outside or
90
101
  inside of the JIT-compiled block for `mpi4py` and `numba-mpi`, respectively.
91
102
  Timing is repeated `N_REPEAT` times and the minimum time is reported.
92
103
  The generated plot shown below depicts the speedup obtained by replacing `mpi4py`
93
- with `numba_mpi`, plotted as a function of `N_TIMES / n_intervals` - the number of MPI calls per
104
+ with `numba_mpi`, plotted as a function of `N_TIMES / n_intervals` - the number of MPI calls per
94
105
  interval. The speedup, which stems from avoiding roundtrips between JIT-compiled
95
106
  and Python code is significant (150%-300%) in all cases. The more often communication
96
- is needed (smaller `n_intervals`), the larger the measured speedup. Note that nothing
107
+ is needed (smaller `n_intervals`), the larger the measured speedup. Note that nothing
97
108
  in the actual number crunching (within the `get_pi_part()` function) or in the employed communication logic
98
109
  (handled by the same MPI library) differs between the `mpi4py` or `numba-mpi` solutions.
99
- These are the overhead of `mpi4py` higher-level abstractions and the overhead of
110
+ These are the overhead of `mpi4py` higher-level abstractions and the overhead of
100
111
  repeatedly entering and leaving the JIT-compiled block if using `mpi4py`, which can be
101
112
  eliminated by using `numba-mpi`, and which the measured differences in execution time
102
113
  stem from.
@@ -156,6 +167,14 @@ if numba_mpi.rank() == 0:
156
167
 
157
168
  ![plot](https://github.com/numba-mpi/numba-mpi/releases/download/tip/readme_plot.png)
158
169
 
170
+ ### Known Issues
171
+
172
+ **NOTE**: Issues listed below only relate to combinations of platforms and MPI distributions that we target to support, but due to various reason are currently not working and are temporarily excluded from automated testing:
173
+
174
+ - tests on Ubuntu 2024.4 that use MPICH are not run due to failures caused by newer version of MPICH (`4.2.0`); note, that previous tests ran
175
+ using version `4.0.2` of MPICH (that is installed by default on Ubuntu 2022.4 using `apt`) were passing (see [related issue](https://github.com/numba-mpi/numba-mpi/issues/162) - TODO #162),
176
+ - tests on Intel MacOS (v13) that use OpenMPI are currently not run due to failures being under investigation (see [related issue](https://github.com/numba-mpi/numba-mpi/issues/163) - TODO #163),
177
+ - `numba-mpi` currently does not support ARM-based MacOS, due to required code improvement (see [related issue](https://github.com/numba-mpi/numba-mpi/issues/164) - TODO #164).
159
178
 
160
179
  ### MPI resources on the web:
161
180
 
@@ -177,11 +196,10 @@ if numba_mpi.rank() == 0:
177
196
 
178
197
  ### Acknowledgements:
179
198
 
180
- We thank [all contributors](https://github.com/numba-mpi/numba-mpi/graphs/contributors) and users who reported feedback to the project
199
+ We thank [all contributors](https://github.com/numba-mpi/numba-mpi/graphs/contributors) and users who reported feedback to the project
181
200
  through [GitHub issues](https://github.com/numba-mpi/numba-mpi/issues).
182
201
 
183
202
  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),
184
- the [Max Planck Society](https://www.mpg.de/en) and the [European Union](https://erc.europa.eu/) (ERC, EmulSim, 101044662).
185
- We further acknowledge Poland’s high-performance computing infrastructure [PLGrid](https://plgrid.pl) (HPC Centers: [ACK Cyfronet AGH](https://www.cyfronet.pl/en))
203
+ the [Max Planck Society](https://www.mpg.de/en) and the [European Union](https://erc.europa.eu/) (ERC, EmulSim, 101044662).
204
+ We further acknowledge Poland’s high-performance computing infrastructure [PLGrid](https://plgrid.pl) (HPC Centers: [ACK Cyfronet AGH](https://www.cyfronet.pl/en))
186
205
  for providing computer facilities and support within computational grant no. PLG/2023/016369.
187
-
@@ -14,31 +14,32 @@
14
14
  [![DOI](https://zenodo.org/badge/316911228.svg)](https://zenodo.org/badge/latestdoi/316911228)
15
15
 
16
16
  ### Overview
17
- 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).
17
+ 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 paper in SoftwareX (open access)](https://www.sciencedirect.com/science/article/pii/S235271102400267X) (please cite if numba-mpi is used in your research).
18
18
 
19
19
  Support is provided for a subset of MPI routines covering: `size`/`rank`, `send`/`recv`, `allreduce`, `reduce`, `bcast`, `scatter`/`gather` & `allgather`, `barrier`, `wtime`
20
20
  and basic asynchronous communication with `isend`/`irecv` (only for contiguous arrays); for request handling including `wait`/`waitall`/`waitany` and `test`/`testall`/`testany`.
21
21
 
22
- The API uses NumPy and supports both numeric and character datatypes (e.g., `broadcast`).
22
+ The API uses NumPy and supports both numeric and character datatypes (e.g., `broadcast`).
23
23
  Auto-generated docstring-based API docs are published on the web: https://numba-mpi.github.io/numba-mpi
24
24
 
25
- Packages can be obtained from
26
- [PyPI](https://pypi.org/project/numba-mpi),
27
- [Conda Forge](https://anaconda.org/conda-forge/numba-mpi),
25
+ Packages can be obtained from
26
+ [PyPI](https://pypi.org/project/numba-mpi),
27
+ [Conda Forge](https://anaconda.org/conda-forge/numba-mpi),
28
28
  [Arch Linux](https://aur.archlinux.org/packages/python-numba-mpi)
29
29
  or by invoking `pip install git+https://github.com/numba-mpi/numba-mpi.git`.
30
30
 
31
31
  numba-mpi is a pure-Python package.
32
32
  The codebase includes a test suite used through the GitHub Actions workflows ([thanks to mpi4py's setup-mpi](https://github.com/mpi4py/setup-mpi)!)
33
- for automated testing on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/)
34
- & [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)),
35
- macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and
36
- Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi)).
33
+ for automated testing on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/)
34
+ & [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)),
35
+ macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and
36
+ Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi)). Note, that some of those
37
+ combinations may not be fully supported yet - see [Known Issues](#known-issues) for more information.
37
38
 
38
39
  Features that are not implemented yet include (help welcome!):
39
40
  - support for non-default communicators
40
41
  - support for `MPI_IN_PLACE` in `[all]gather`/`scatter` and `allreduce`
41
- - support for `MPI_Type_create_struct` (Numpy structured arrays)
42
+ - support for `MPI_Type_create_struct` (Numpy structured arrays)
42
43
  - ...
43
44
 
44
45
  ### Hello world send/recv example:
@@ -61,21 +62,21 @@ hello()
61
62
  ### Example comparing numba-mpi vs. mpi4py performance:
62
63
 
63
64
  The example below compares `Numba`+`mpi4py` vs. `Numba`+`numba-mpi` performance.
64
- The sample code estimates $\pi$ by numerical integration of $\int_0^1 (4/(1+x^2))dx=\pi$
65
- dividing the workload into `n_intervals` handled by separate MPI processes
65
+ The sample code estimates $\pi$ by numerical integration of $\int_0^1 (4/(1+x^2))dx=\pi$
66
+ dividing the workload into `n_intervals` handled by separate MPI processes
66
67
  and then obtaining a sum using `allreduce` (see, e.g., analogous [Matlab docs example](https://www.mathworks.com/help/parallel-computing/numerical-estimation-of-pi-using-message-passing.html)).
67
68
  The computation is carried out in a JIT-compiled function `get_pi_part()` and is repeated
68
- `N_TIMES`. The repetitions and the MPI-handled reduction are done outside or
69
+ `N_TIMES`. The repetitions and the MPI-handled reduction are done outside or
69
70
  inside of the JIT-compiled block for `mpi4py` and `numba-mpi`, respectively.
70
71
  Timing is repeated `N_REPEAT` times and the minimum time is reported.
71
72
  The generated plot shown below depicts the speedup obtained by replacing `mpi4py`
72
- with `numba_mpi`, plotted as a function of `N_TIMES / n_intervals` - the number of MPI calls per
73
+ with `numba_mpi`, plotted as a function of `N_TIMES / n_intervals` - the number of MPI calls per
73
74
  interval. The speedup, which stems from avoiding roundtrips between JIT-compiled
74
75
  and Python code is significant (150%-300%) in all cases. The more often communication
75
- is needed (smaller `n_intervals`), the larger the measured speedup. Note that nothing
76
+ is needed (smaller `n_intervals`), the larger the measured speedup. Note that nothing
76
77
  in the actual number crunching (within the `get_pi_part()` function) or in the employed communication logic
77
78
  (handled by the same MPI library) differs between the `mpi4py` or `numba-mpi` solutions.
78
- These are the overhead of `mpi4py` higher-level abstractions and the overhead of
79
+ These are the overhead of `mpi4py` higher-level abstractions and the overhead of
79
80
  repeatedly entering and leaving the JIT-compiled block if using `mpi4py`, which can be
80
81
  eliminated by using `numba-mpi`, and which the measured differences in execution time
81
82
  stem from.
@@ -135,6 +136,14 @@ if numba_mpi.rank() == 0:
135
136
 
136
137
  ![plot](https://github.com/numba-mpi/numba-mpi/releases/download/tip/readme_plot.png)
137
138
 
139
+ ### Known Issues
140
+
141
+ **NOTE**: Issues listed below only relate to combinations of platforms and MPI distributions that we target to support, but due to various reason are currently not working and are temporarily excluded from automated testing:
142
+
143
+ - tests on Ubuntu 2024.4 that use MPICH are not run due to failures caused by newer version of MPICH (`4.2.0`); note, that previous tests ran
144
+ using version `4.0.2` of MPICH (that is installed by default on Ubuntu 2022.4 using `apt`) were passing (see [related issue](https://github.com/numba-mpi/numba-mpi/issues/162) - TODO #162),
145
+ - tests on Intel MacOS (v13) that use OpenMPI are currently not run due to failures being under investigation (see [related issue](https://github.com/numba-mpi/numba-mpi/issues/163) - TODO #163),
146
+ - `numba-mpi` currently does not support ARM-based MacOS, due to required code improvement (see [related issue](https://github.com/numba-mpi/numba-mpi/issues/164) - TODO #164).
138
147
 
139
148
  ### MPI resources on the web:
140
149
 
@@ -156,11 +165,10 @@ if numba_mpi.rank() == 0:
156
165
 
157
166
  ### Acknowledgements:
158
167
 
159
- We thank [all contributors](https://github.com/numba-mpi/numba-mpi/graphs/contributors) and users who reported feedback to the project
168
+ We thank [all contributors](https://github.com/numba-mpi/numba-mpi/graphs/contributors) and users who reported feedback to the project
160
169
  through [GitHub issues](https://github.com/numba-mpi/numba-mpi/issues).
161
170
 
162
171
  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),
163
- the [Max Planck Society](https://www.mpg.de/en) and the [European Union](https://erc.europa.eu/) (ERC, EmulSim, 101044662).
164
- We further acknowledge Poland’s high-performance computing infrastructure [PLGrid](https://plgrid.pl) (HPC Centers: [ACK Cyfronet AGH](https://www.cyfronet.pl/en))
172
+ the [Max Planck Society](https://www.mpg.de/en) and the [European Union](https://erc.europa.eu/) (ERC, EmulSim, 101044662).
173
+ We further acknowledge Poland’s high-performance computing infrastructure [PLGrid](https://plgrid.pl) (HPC Centers: [ACK Cyfronet AGH](https://www.cyfronet.pl/en))
165
174
  for providing computer facilities and support within computational grant no. PLG/2023/016369.
166
-
@@ -1,5 +1,5 @@
1
- """ Utilities for handling MPI_Requests with respective Wait and Test function
2
- wrappers.
1
+ """Utilities for handling MPI_Requests with respective Wait and Test function
2
+ wrappers.
3
3
  """
4
4
 
5
5
  import ctypes
@@ -1,6 +1,6 @@
1
- Metadata-Version: 2.1
1
+ Metadata-Version: 2.2
2
2
  Name: numba-mpi
3
- Version: 1.1.1
3
+ Version: 1.1.3
4
4
  Summary: Numba @jittable MPI wrappers tested on Linux, macOS and Windows
5
5
  Home-page: https://github.com/numba-mpi/numba-mpi
6
6
  Author: https://github.com/numba-mpi/numba-mpi/graphs/contributors
@@ -18,6 +18,16 @@ Requires-Dist: psutil
18
18
  Provides-Extra: tests
19
19
  Requires-Dist: pytest<8.0.0; extra == "tests"
20
20
  Requires-Dist: py-pde; extra == "tests"
21
+ Dynamic: author
22
+ Dynamic: description
23
+ Dynamic: description-content-type
24
+ Dynamic: home-page
25
+ Dynamic: license
26
+ Dynamic: project-url
27
+ Dynamic: provides-extra
28
+ Dynamic: requires-dist
29
+ Dynamic: requires-python
30
+ Dynamic: summary
21
31
 
22
32
  # <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
23
33
 
@@ -35,31 +45,32 @@ Requires-Dist: py-pde; extra == "tests"
35
45
  [![DOI](https://zenodo.org/badge/316911228.svg)](https://zenodo.org/badge/latestdoi/316911228)
36
46
 
37
47
  ### Overview
38
- 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).
48
+ 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 paper in SoftwareX (open access)](https://www.sciencedirect.com/science/article/pii/S235271102400267X) (please cite if numba-mpi is used in your research).
39
49
 
40
50
  Support is provided for a subset of MPI routines covering: `size`/`rank`, `send`/`recv`, `allreduce`, `reduce`, `bcast`, `scatter`/`gather` & `allgather`, `barrier`, `wtime`
41
51
  and basic asynchronous communication with `isend`/`irecv` (only for contiguous arrays); for request handling including `wait`/`waitall`/`waitany` and `test`/`testall`/`testany`.
42
52
 
43
- The API uses NumPy and supports both numeric and character datatypes (e.g., `broadcast`).
53
+ The API uses NumPy and supports both numeric and character datatypes (e.g., `broadcast`).
44
54
  Auto-generated docstring-based API docs are published on the web: https://numba-mpi.github.io/numba-mpi
45
55
 
46
- Packages can be obtained from
47
- [PyPI](https://pypi.org/project/numba-mpi),
48
- [Conda Forge](https://anaconda.org/conda-forge/numba-mpi),
56
+ Packages can be obtained from
57
+ [PyPI](https://pypi.org/project/numba-mpi),
58
+ [Conda Forge](https://anaconda.org/conda-forge/numba-mpi),
49
59
  [Arch Linux](https://aur.archlinux.org/packages/python-numba-mpi)
50
60
  or by invoking `pip install git+https://github.com/numba-mpi/numba-mpi.git`.
51
61
 
52
62
  numba-mpi is a pure-Python package.
53
63
  The codebase includes a test suite used through the GitHub Actions workflows ([thanks to mpi4py's setup-mpi](https://github.com/mpi4py/setup-mpi)!)
54
- for automated testing on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/)
55
- & [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)),
56
- macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and
57
- Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi)).
64
+ for automated testing on: Linux ([MPICH](https://www.mpich.org/), [OpenMPI](https://www.open-mpi.org/doc/)
65
+ & [Intel MPI](https://www.intel.com/content/www/us/en/developer/tools/oneapi/mpi-library.html)),
66
+ macOS ([MPICH](https://www.mpich.org/) & [OpenMPI](https://www.open-mpi.org/doc/)) and
67
+ Windows ([MS MPI](https://docs.microsoft.com/en-us/message-passing-interface/microsoft-mpi)). Note, that some of those
68
+ combinations may not be fully supported yet - see [Known Issues](#known-issues) for more information.
58
69
 
59
70
  Features that are not implemented yet include (help welcome!):
60
71
  - support for non-default communicators
61
72
  - support for `MPI_IN_PLACE` in `[all]gather`/`scatter` and `allreduce`
62
- - support for `MPI_Type_create_struct` (Numpy structured arrays)
73
+ - support for `MPI_Type_create_struct` (Numpy structured arrays)
63
74
  - ...
64
75
 
65
76
  ### Hello world send/recv example:
@@ -82,21 +93,21 @@ hello()
82
93
  ### Example comparing numba-mpi vs. mpi4py performance:
83
94
 
84
95
  The example below compares `Numba`+`mpi4py` vs. `Numba`+`numba-mpi` performance.
85
- The sample code estimates $\pi$ by numerical integration of $\int_0^1 (4/(1+x^2))dx=\pi$
86
- dividing the workload into `n_intervals` handled by separate MPI processes
96
+ The sample code estimates $\pi$ by numerical integration of $\int_0^1 (4/(1+x^2))dx=\pi$
97
+ dividing the workload into `n_intervals` handled by separate MPI processes
87
98
  and then obtaining a sum using `allreduce` (see, e.g., analogous [Matlab docs example](https://www.mathworks.com/help/parallel-computing/numerical-estimation-of-pi-using-message-passing.html)).
88
99
  The computation is carried out in a JIT-compiled function `get_pi_part()` and is repeated
89
- `N_TIMES`. The repetitions and the MPI-handled reduction are done outside or
100
+ `N_TIMES`. The repetitions and the MPI-handled reduction are done outside or
90
101
  inside of the JIT-compiled block for `mpi4py` and `numba-mpi`, respectively.
91
102
  Timing is repeated `N_REPEAT` times and the minimum time is reported.
92
103
  The generated plot shown below depicts the speedup obtained by replacing `mpi4py`
93
- with `numba_mpi`, plotted as a function of `N_TIMES / n_intervals` - the number of MPI calls per
104
+ with `numba_mpi`, plotted as a function of `N_TIMES / n_intervals` - the number of MPI calls per
94
105
  interval. The speedup, which stems from avoiding roundtrips between JIT-compiled
95
106
  and Python code is significant (150%-300%) in all cases. The more often communication
96
- is needed (smaller `n_intervals`), the larger the measured speedup. Note that nothing
107
+ is needed (smaller `n_intervals`), the larger the measured speedup. Note that nothing
97
108
  in the actual number crunching (within the `get_pi_part()` function) or in the employed communication logic
98
109
  (handled by the same MPI library) differs between the `mpi4py` or `numba-mpi` solutions.
99
- These are the overhead of `mpi4py` higher-level abstractions and the overhead of
110
+ These are the overhead of `mpi4py` higher-level abstractions and the overhead of
100
111
  repeatedly entering and leaving the JIT-compiled block if using `mpi4py`, which can be
101
112
  eliminated by using `numba-mpi`, and which the measured differences in execution time
102
113
  stem from.
@@ -156,6 +167,14 @@ if numba_mpi.rank() == 0:
156
167
 
157
168
  ![plot](https://github.com/numba-mpi/numba-mpi/releases/download/tip/readme_plot.png)
158
169
 
170
+ ### Known Issues
171
+
172
+ **NOTE**: Issues listed below only relate to combinations of platforms and MPI distributions that we target to support, but due to various reason are currently not working and are temporarily excluded from automated testing:
173
+
174
+ - tests on Ubuntu 2024.4 that use MPICH are not run due to failures caused by newer version of MPICH (`4.2.0`); note, that previous tests ran
175
+ using version `4.0.2` of MPICH (that is installed by default on Ubuntu 2022.4 using `apt`) were passing (see [related issue](https://github.com/numba-mpi/numba-mpi/issues/162) - TODO #162),
176
+ - tests on Intel MacOS (v13) that use OpenMPI are currently not run due to failures being under investigation (see [related issue](https://github.com/numba-mpi/numba-mpi/issues/163) - TODO #163),
177
+ - `numba-mpi` currently does not support ARM-based MacOS, due to required code improvement (see [related issue](https://github.com/numba-mpi/numba-mpi/issues/164) - TODO #164).
159
178
 
160
179
  ### MPI resources on the web:
161
180
 
@@ -177,11 +196,10 @@ if numba_mpi.rank() == 0:
177
196
 
178
197
  ### Acknowledgements:
179
198
 
180
- We thank [all contributors](https://github.com/numba-mpi/numba-mpi/graphs/contributors) and users who reported feedback to the project
199
+ We thank [all contributors](https://github.com/numba-mpi/numba-mpi/graphs/contributors) and users who reported feedback to the project
181
200
  through [GitHub issues](https://github.com/numba-mpi/numba-mpi/issues).
182
201
 
183
202
  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),
184
- the [Max Planck Society](https://www.mpg.de/en) and the [European Union](https://erc.europa.eu/) (ERC, EmulSim, 101044662).
185
- We further acknowledge Poland’s high-performance computing infrastructure [PLGrid](https://plgrid.pl) (HPC Centers: [ACK Cyfronet AGH](https://www.cyfronet.pl/en))
203
+ the [Max Planck Society](https://www.mpg.de/en) and the [European Union](https://erc.europa.eu/) (ERC, EmulSim, 101044662).
204
+ We further acknowledge Poland’s high-performance computing infrastructure [PLGrid](https://plgrid.pl) (HPC Centers: [ACK Cyfronet AGH](https://www.cyfronet.pl/en))
186
205
  for providing computer facilities and support within computational grant no. PLG/2023/016369.
187
-
@@ -1,4 +1,4 @@
1
- """ the magick behind ``pip install ...`` """
1
+ """the magick behind ``pip install ...``"""
2
2
 
3
3
  from setuptools import find_packages, setup
4
4
 
@@ -18,7 +18,7 @@ setup(
18
18
  url="https://github.com/numba-mpi/numba-mpi",
19
19
  author="https://github.com/numba-mpi/numba-mpi/graphs/contributors",
20
20
  use_scm_version={
21
- "local_scheme": lambda _: "",
21
+ "local_scheme": "no-local-version",
22
22
  "version_scheme": "post-release",
23
23
  },
24
24
  python_requires=">=3.8",
@@ -1,4 +1,4 @@
1
- """ checks if listings from the paper run OK """
1
+ """checks if listings from the paper run OK"""
2
2
 
3
3
  import pathlib
4
4
 
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes
File without changes