MPI通信パターン

Question

これを行うにはスマートな方法があるのだろうかと思っていました。私が3つのノード0,1,2を持っているとしましょう。そして、各ノードに配列a0、a1、a2があるとしましょう。各ノードの内容は

a0 = {0, 1, 2, 1} 
a1 = {1, 2, 2, 0} 
a2 = {0, 0, 1, 2} 

のようなものが巧妙な通信パターンが存在している場合ので、それはすなわち、ノード

a0 = {0, 0, 0, 0} 
a1 = {1, 1, 1, 1} 
a2 = {2, 2, 2, 2} 

私が考えているアプローチを対応だし、各番号を移動するには、並べ替え伴うだろうと一時的なバッファーが、私は賢い方法があったかどうか疑問に思っていた？

Answer 1

次のようにして、このためMPI_Alltoallvを使用することができます。

1. ソートlocal_data（a）昇順で各要素の節点に対応することによって。
2. send_displacements[r]が、ノードrを参照するlocal_dataの最初の要素のインデックスを示すように、send_displacementsの配列を作成します。
3. send_counts[r]がノードrに対応するlocal_dataの要素の数に等しくなるようにsend_countsの配列を作成します。これは最後のランクを除いてsend_counts[r] = send_displacements[r+1] - send_displacements[r]と計算できます。
4. MPI_Alltoall(send_counts, 1, MPI_INT, recv_counts, 1, MPI_INT, comm)
5. recv_displacements[r] = sum(recv_counts[r'] for all r' < r)のように計算する。
6. recv_dataにsum（recv_counts）要素を用意します。
7. MPI_Alltoallv(local_data, send_counts, send_displacements, MPI_INT, recv_data, recv_counts, recv_displacements, MPI_INT, comm)