Skip to content

Commit 7939587

Browse files
committed
minor: doc update with MPIMatrixMult
1 parent 256b98e commit 7939587

File tree

2 files changed

+52
-25
lines changed

2 files changed

+52
-25
lines changed

Makefile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ doc_nccl:
7878
rm tutorials/*_cupy.py tutorials/*_nccl.py
7979

8080
docupdate:
81-
cd docs && make html && cd ..
81+
cd docs && NCCL_PYLOPS_MPI=0 make html && cd ..
8282

8383
servedoc:
8484
$(PYTHON) -m http.server --directory docs/build/

docs/source/gpu.rst

Lines changed: 51 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -120,41 +120,68 @@ one MPI process. In fact, minor communications like those dealing with array-rel
120120
The CuPy and NCCL backend is in active development, with many examples not yet in the docs.
121121
You can find many `other examples <https://github.com/PyLops/pylops_notebooks/tree/master/developement-mpi/Cupy_MPI>`_ from the `PyLops Notebooks repository <https://github.com/PyLops/pylops_notebooks>`_.
122122

123-
Supports for NCCL Backend
124-
----------------------------
125-
In the following, we provide a list of modules (i.e., operators and solvers) where we plan to support NCCL and the current status:
123+
Supports for CuPy and NCCL
124+
--------------------------
125+
In the following, we provide a list of modules (i.e., operators and solvers) with their current status (available on CPU+MPI,
126+
GPU+MPI, and GPU+NCCL):
126127

127128
.. list-table::
128-
:widths: 50 25
129+
:widths: 50 25 25 25
129130
:header-rows: 1
130131

131-
* - modules
132-
- NCCL supported
132+
* - Operator/method
133+
- CPU
134+
- GPU+MPI
135+
- GPU+NCCL
133136
* - :class:`pylops_mpi.DistributedArray`
134-
- ✅
135-
* - :class:`pylops_mpi.basicoperators.MPIVStack`
136-
- ✅
137+
- ✅
138+
- ✅
139+
- ✅
140+
* - :class:`pylops_mpi.basicoperators.MPIMatrixMult`
141+
- ✅
142+
- 🔴
143+
- 🔴
137144
* - :class:`pylops_mpi.basicoperators.MPIVStack`
138-
- ✅
145+
- ✅
146+
- ✅
147+
- ✅
139148
* - :class:`pylops_mpi.basicoperators.MPIHStack`
140-
- ✅
149+
- ✅
150+
- ✅
151+
- ✅
141152
* - :class:`pylops_mpi.basicoperators.MPIBlockDiag`
142-
- ✅
143-
* - :class:`pylops_mpi.basicoperators.MPIGradient`
144-
- ✅
153+
- ✅
154+
- ✅
155+
- ✅
145156
* - :class:`pylops_mpi.basicoperators.MPIFirstDerivative`
157+
- ✅
158+
- ✅
146159
- ✅
147160
* - :class:`pylops_mpi.basicoperators.MPISecondDerivative`
148-
- ✅
161+
- ✅
162+
- ✅
163+
- ✅
149164
* - :class:`pylops_mpi.basicoperators.MPILaplacian`
150-
- ✅
165+
- ✅
166+
- ✅
167+
- ✅
168+
* - :class:`pylops_mpi.basicoperators.MPIGradient`
169+
- ✅
170+
- ✅
171+
- ✅
172+
* - :class:`pylops_mpi.signalprocessing.MPIFredhoml1`
173+
- ✅
174+
- ✅
175+
- ✅
176+
* - :class:`pylops_mpi.waveeqprocessing.MPIMDC`
177+
- ✅
178+
- ✅
179+
- ✅
151180
* - :class:`pylops_mpi.optimization.basic.cg`
152-
- ✅
181+
- ✅
182+
- ✅
183+
- ✅
153184
* - :class:`pylops_mpi.optimization.basic.cgls`
154-
- ✅
155-
* - :class:`pylops_mpi.signalprocessing.Fredhoml1`
156-
- ✅
157-
* - Complex Numeric Data Type for NCCL
158-
- ✅
159-
* - ISTA Solver
160-
- Planned ⏳
185+
- ✅
186+
- ✅
187+
- ✅

0 commit comments

Comments
 (0)