Fortranの「行列転置」で涙を流さないために：キャッシュとメモリレイアウトの深淵
こんにちは。かつて宇宙の深淵を数値で追いかけていた、ただの「Fortranオタク」です。
C言語やPythonから流れてきた皆さんが、Fortranを触って最初に驚くこと、それは「行列計算の圧倒的な速さ」と「書き方一つで劇的に遅くなる落とし穴」の二面性ではないでしょうか。今日は、Fortran初学者の多くが最初にぶつかる壁、「行列の転置とキャッシュ効率」について、現場の泥臭い知見を交えてお話しします。
—
1. Fortranは「列優先」という絶対ルールで生きている
まず、Fortranのメモリ配置を理解しましょう。Python（NumPy）やC言語は「行優先（Row-major）」ですが、Fortranは「列優先（Column-major）」です。
これはどういうことか？ 2次元配列 `A(M, N)` をメモリ上に展開したとき、`A(1, 1), A(2, 1), A(3, 1)…` という順序で並んでいるということです。
- 行優先（C/Python）: `A(i, j)` と `A(i, j+1)` が隣接している。
- 列優先（Fortran）: `A(i, j)` と `A(i+1, j)` が隣接している。
現代のCPUは、メモリからデータを読み込む際、隣接するデータをごっそりとキャッシュメモリに溜め込みます。つまり、「メモリ上で隣り合っているデータ」を順番に触るのが、高速化の鉄則なんです。
—
2. TRANSPOSE関数は「魔法」か「罠」か
さて、行列を転置したいとき、Fortranには便利な組込み関数 `TRANSPOSE` があります。
! シンプルな転置
B = TRANSPOSE(A)
これ、一見すると便利ですよね。でも、現場のシミュレーションでは「ちょっと待て」と言いたい。なぜなら、`TRANSPOSE` は新しいメモリ領域を確保し、値をコピーする処理を内部で行うからです。
もし、計算のループの中で毎回 `TRANSPOSE` を呼び出していたらどうなるか。メモリの確保とコピーという「重い処理」が走るだけでなく、その結果として、せっかくCPUのキャッシュに乗っていたデータが強制的に追い出されてしまいます。
—
3. 「転置」させずに「アクセス順」を変える
プロの現場では、可能な限り `TRANSPOSE` 関数を使いません。代わりに、「転置されたものとしてループの添字を工夫する」のが定石です。
例えば、行列 $A$ の転置 $A^T$ を使った行列積を計算したいとき、安易に転置してから計算するのではなく、「添字を入れ替えてループを回す」ことで、メモリを列優先のルール通りに読み込ませるのです。
! ダメな例：転置してから計算（キャッシュミス連発）
B = TRANSPOSE(A)
do j = 1, N
do i = 1, M
val = B(i, j) C(i, j)
! …
end do
end do
! 良い例：転置せず、添字を工夫して列優先を守る
do j = 1, N
do i = 1, M
! A(j, i) にアクセスすることで、メモリの連続性を保つ
val = A(j, i) C(i, j)
! …
end do
end do
このように、メモリ上の並びを意識した添字の回し方をするだけで、大規模なシミュレーションでは計算時間が数倍〜数十倍変わることも珍しくありません。
—
4. コンパイラはあなたの味方か？
最後に、現場で生き残るための「コンパイラフラグ」のコツを授けます。`gfortran` や `ifort (ifx)` を使う際、最適化オプションを使い分けるのが重要です。
- `-O3`: 基本ですが、ここからがスタートラインです。
- `-march=native`: コンパイルしているCPUの限界まで命令セット（AVX-512など）を使い倒します。
- `-fopt-info-vec`: コンパイラが「どこをベクトル化（高速化）できたか」を教えてくれます。これを見て、ループが最適化されているか確認してください。
もし、「ループの順番をどう頑張っても列優先に合わせられない…」という場合は、コンパイラに「この配列は転置して扱いたい」と伝えるための「インターチェンジ（Loop Interchange）」という最適化手法をコンパイラにヒントとして与えることもあります。
—
まとめ：まずは「メモリの並び」を想像しよう
Fortranを書くときは、コードを書く前に「今、メモリのどの場所を触っているか？」を頭の中でイメージしてください。
1. `TRANSPOSE` は便利だが、ループの外で一度だけ使うもの。
2. ループの中では、メモリの並び（列優先）を意識した添字の順番にする。
3. コンパイラの最適化レポートを見て、自分のコードが期待通り動いているか確認する。
これだけで、あなたの書くプログラムは、世界中のスーパーコンピュータで輝く一級品のコードに変わります。Fortranは古臭い言語ではありません。メモリを極限までコントロールできる、最高の「精密機械」なんです。
さあ、まずは手元のコードで、ループの添字を一つ入れ替えてみてください。その瞬間に、計算機が軽やかに唸り始めるのを感じられるはずですよ！応援しています！
コメント