From 99f6f4b1b4fd9397656c76bfd7b0743b2fad80cc Mon Sep 17 00:00:00 2001
From: Claudia Comito <39374113+ClaudiaComito@users.noreply.github.com>
Date: Mon, 28 Oct 2024 17:45:43 +0100
Subject: [PATCH] add Dalcin et al reference (#1695)

---
 heat/core/manipulations.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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)