diff --git a/doc/requirements.txt b/doc/requirements.txt index 861a337bf..672d02096 100644 --- a/doc/requirements.txt +++ b/doc/requirements.txt @@ -1,5 +1,5 @@ Sphinx==8.1.3 -sphinx-autoapi===3.3.2 +sphinx-autoapi===3.3.3 sphinx_rtd_theme==3.0.1 sphinxcontrib-napoleon==0.7 sphinx-copybutton==0.5.2 diff --git a/heat/core/manipulations.py b/heat/core/manipulations.py index 49c67469e..02ec09ec9 100644 --- a/heat/core/manipulations.py +++ b/heat/core/manipulations.py @@ -3719,7 +3719,8 @@ def _axis2axisResplit( comm: Communication, ) -> torch.Tensor: """ - Resplits the input array along a new axis and performs data exchange using MPI_Alltoallw. Returns target_larray object with the data after the exchange. + Resplits the input array along a new axis and performs data exchange using MPI_Alltoallw, after [1]. + Returns `target_larray` object with the data after the exchange. Parameters ---------- @@ -3737,6 +3738,10 @@ def _axis2axisResplit( The tiling object containing the subarray parameters for the target array. comm : Communication The communication object for MPI communication. + + References + ---------- + [1] Dalcin, Mortensen, Keyes, "Fast parallel multidimensional FFT using advanced MPI", 2018. """ # Create subarray types for original local shapes split along the new axis source_subarray_params = source_tiles.get_subarray_params(source_split, target_split)