6.2. MPI並列化

本節では \({\mathcal H}\Phi\) で使用されるMPI並列化戦略について説明します。

6.2.1. 概要

\({\mathcal H}\Phi\) はMPIを使用して複数プロセス間で計算を並列化します。 ヒルベルト空間の基底状態はMPIプロセス間に分散され、 ハミルトニアンの行列ベクトル積 \(\hat{H}|\psi\rangle\) が並列に実行されます。

6.2.2. サイトの分類

\(N\) サイト系で \(N_{\rm proc}\) MPIプロセスを使用する場合、 サイトは2つのカテゴリに分類されます:

ローカルサイト (\(N_{\rm local}\) サイト)

量子状態が各プロセス内に格納されるサイト。 これらのサイトに対する基底状態操作はローカルで実行可能。

プロセス間サイト (\(N_{\rm inter}\) サイト)

量子状態が複数プロセスにまたがるサイト。 これらのサイトに対する操作にはMPI通信が必要。

関係式:

(6.1)\[N_{\rm local} = N - \log_2(N_{\rm proc})\]
(6.2)\[N_{\rm inter} = \log_2(N_{\rm proc})\]

例えば、8サイト系で4 MPIプロセス(\(N_{\rm proc}=4\))を使用する場合、 ローカルサイト \(N_{\rm local}=6\)、プロセス間サイト \(N_{\rm inter}=2\) となります。

6.2.3. MPI通信パターン

ハミルトニアン中の演算は、必要なMPI通信によって分類されます:

ローカル演算

演算に関わる両サイトがローカルサイト。 MPI通信は不要。

MPIsingle演算

一方がローカルサイト、他方がプロセス間サイト。 1つのパートナープロセスとのMPI通信が必要。 通信相手(origin)は以下で決定:

(6.3)\[\text{origin} = \text{myrank} \oplus T_{\rm pow}[\text{site}]\]

ここで \(\oplus\) はXOR演算、\(T_{\rm pow}\) は各サイトの2のべき乗です。

MPIdouble演算

両サイトがプロセス間サイト。 MPI通信が必要だが、受信データは一様に処理可能。 通信相手は:

(6.4)\[\text{origin} = \text{myrank} \oplus (T_{\rm pow}[\text{site}_1] + T_{\rm pow}[\text{site}_2])\]

6.2.4. バッチ処理MPI通信

MPI通信オーバーヘッドを削減するため、\({\mathcal H}\Phi\) は 同じ通信相手(origin)を持つ複数の演算をバッチにグループ化します:

最適化前:

プロセス間サイトを持つ各転送項/相互作用項が個別に MPI_Sendrecv を呼び出す。 同じoriginを持つ \(N\) 項がある場合、\(N\) 回の独立した通信が発生。

最適化後:

同じoriginを持つ項をグループ化。 1回の MPI_Sendrecv で必要なデータを取得し、 その後 \(N\) 項すべてをローカルで処理。

この最適化は以下の場合に特に効果的:

  • 多くのホッピング項が同じ通信相手を共有する2D/3D格子モデル

  • 隣接サイト間に多くの転送項を持つ系

バッチ通信は以下のモデルに実装済み:

  • SpinlessFermionモデル(MPIsingle)

  • Hubbardモデル(MPIsingle、MPIdouble;InterAllはHubbardGCのみ)

  • Spinモデル(Exchange MPIsingle;SpinGCはPairLiftも一括処理)

MPIビルドでは、バッチ通信はデフォルトで有効です。入力ファイルでの指定は不要です。 この最適化は、対応しているプロセス間項の通信・適用経路を変更しますが、 従来の項ごとのMPI通信経路と同じ数値結果を再現するよう設計されています。

注釈

時間発展(TimeEvolution)モードでのバッチ通信の制限

TETwoBody または TEOneBody``(expert mode step 依存型、 ``NTEInterAllMax > 0 もしくは NTETransferMax > 0)を使用する 時間発展計算では、MPI バッチ通信は自動的に無効化され、 従来の per-term MPI 通信にフォールバックします。

これは、バッチ通信のグループ構造がプロセス起動時に 1 回だけ初期化される 設計のため、step ごとに追加・変更される相互作用項を追従できないためです。

Peierls 代用(AC Laser モード、PumpType = "AC Laser")は 既存の転送項の係数のみを変更するため、バッチ通信の恩恵を受けられます。

バッチ通信の無効化

環境変数 HPHI_MPI_NOBATCH=1 を設定すると、MPI通信のバッチ処理を全体で無効化し、 バッチ化前の項ごとのMPI通信にフォールバックします。デバッグや、バッチ最適化が 非バッチ時と同一の結果を再現することの検証に使用します。また、特定の計算機・入力で 従来経路との性能比較を行う場合にも使用できます:

export HPHI_MPI_NOBATCH=1
mpirun -np 4 ./HPhi -e namelist.def

起動時に MPI batching : ON または MPI batching : OFF と表示されます。 この値は rank 0 で読み込まれ全ランクにブロードキャストされるため、 常に全プロセスで一致します。

6.2.5. SpinlessFermionのoff-diagonal two-body Green関数

SpinlessFermionにおける \(\langle c^\dagger_i c_j c^\dagger_k c_l \rangle\) のMPI経路は、グランドカノニカル(GC)に加えてカノニカルでも プロセス間サイトを含む場合に対応しています。

この経路では、グローバルビット表現で演算子列を適用し、 ランク間通信を伴う場合でもフェルミオン符号を一貫して評価します。

6.2.6. MPI前提テストの挙動(ctest)

以下のテストはMPI実行環境を前提とします:

  • mpi_consistency_* (目安: -np >= 2

  • lanczos_*_mpidouble (目安: -np = 16

MPIRUN が未設定、または必要なMPIランク条件を満たさない場合、 これらは ctest 上で Skipped として報告されます。

例:

MPIRUN='mpirun -np 4 --oversubscribe' ctest -L consistency
MPIRUN='mpirun -np 16 --oversubscribe' ctest -L batching

6.2.7. フェルミオン符号

フェルミオン系(Hubbard、SpinlessFermion)では、 演算子の反交換から生じるフェルミオン符号をMPI通信時に注意深く扱う必要があります。

符号は SgnBit 関数を使用して計算され、 生成演算子と消滅演算子の間の占有状態数をカウントします。 この符号は初期化時に事前計算され、各転送グループと共に格納されます。

6.2.8. プロセス数の要件

MPIプロセス数は特定の制約を満たす必要があります:

Hubbard/近藤モデル

プロセス数 = \(4^n\) (1サイトあたり4状態: 空、上、下、二重占有)

スピン1/2モデル

プロセス数 = \(2^n\) (1サイトあたり2状態: 上、下)

一般スピンモデル

プロセス数 = プロセス間サイトの \((2S_i+1)\) の積

プロセス数設定の詳細は プロセス数の設定 を参照してください。