add Dalcin et al reference (#1695)

helmholtz-analytics · Oct 28, 2024 · 99f6f4b · 99f6f4b
1 parent ac392b5
commit 99f6f4b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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)