宇宙開発の現場から：Fortranで「巨大な配列」を爆速で初期化する技術
こんにちは。長年、数値シミュレーションの世界で「1秒の計算時間を削るために寝食を忘れる」ような生活を送ってきた者です。
皆さんは、CやPythonで育ってきたエンジニアですね。Fortranと聞くと「古い言語」というイメージがあるかもしれませんが、実は「メモリの挙動を人間がコントロールできる」という意味で、現代でも最強の計算ツールです。
今日は、Fortranで巨大な数値データを扱う際の最初の壁、「配列の初期化コスト」について、現場の泥臭い知見を交えてお話しします。
—
1. DATA文：静的な「おまじない」の正体
Fortranには古くからある `DATA` 文というものがあります。
real(8) :: array(1000)
data array /10000.0d0/ ! 1000個分、全部0で埋める
これは「プログラムの実行ファイルがメモリにロードされた瞬間（あるいは実行開始直後）」に配置される初期化です。非常に高速ですが、「静的」であるという制約があります。一度決めたサイズから変更できないため、計算途中でサイズを変えたい現代的なプログラムには少し不自由です。
2. ALLOCATE時の初期化コスト：落とし穴はここにある
動的にメモリを確保する `ALLOCATE` を使うとき、皆さんはどうしていますか？
real(8), allocatable :: array(:)
allocate(array(1000000))
! ここで別途ループを回して初期化していませんか？
もしここで `do i = 1, 1000000; array(i) = 0.0d0; end do` と書いているとしたら、それは非常にもったいない！
Fortran 2003以降、`ALLOCATE` には `SOURCE=` という強力なオプションが追加されました。これを使うと、メモリを確保した瞬間に値を詰め込むことができます。
! 高速なゼロクリアの例
allocate(array(1000000), source=0.0d0)
これがなぜ重要かというと、OSのカーネルレベルで「メモリを割り当てる」という処理と「値を書き込む」という処理が、CPUのキャッシュラインを意識した最適化によって、最適に統合される可能性が高まるからです。
—
3. 現場が教える「ゼロクリア」の極意
実は、単純なゼロクリアであれば、Fortranの組み込み関数 `call execute_command_line` を使うようなトリッキーなことをしなくても、`ISO_FORTRAN_ENV` を活用した高度なアプローチや、コンパイラの組み込み関数を信じるのが一番です。
しかし、最も重要なのは「配列のアクセス順序」です。Fortranは「列優先（Column-major）」です。つまり、メモリ上では縦方向に並んでいます。
! 良い例：列優先でアクセスする
do j = 1, N
do i = 1, M
array(i, j) = 0.0d0
end do
end do
もしこれを `do i` を外側にして書くと、CPUの「キャッシュミス」が頻発し、せっかくのモダンな記述も台無しになります。「一番左の添字を一番速く動かす」。これがFortranエンジニアの鉄則です。
—
4. コンパイラへの「指示書」を忘れないで
最後に、いくらコードを綺麗に書いても、コンパイラに「本気を出せ」と言わなければ意味がありません。例えば `gfortran` を使う場合、以下のフラグは必須です。
最適化レベル3、かつCPUのパイプラインを最大活用するフラグ
gfortran -O3 -march=native -funroll-loops my_simulation.f90 -o sim.exe
- `-O3`: 冗長なループを剥がし、SIMD命令（ベクトル演算）を適用する許可を出します。
- `-march=native`: あなたが今使っているPCのCPUの限界性能を引き出すための設定です。
—
若手エンジニアへのエール
Fortranは、一見すると無骨な言語に見えます。しかし、メモリの配置を意識し、キャッシュラインを汚さないように書いたコードが、他の言語を圧倒する速度で計算を終える瞬間……この快感は、数値計算の世界でしか味わえません。
まずは、既存の `do` ループによる初期化を `source=` に書き換えるところから始めてみてください。たったそれだけで、あなたのプログラムは少しだけ、宇宙開発の現場で使われるコードに近づきます。
何かあれば、いつでもコードを見せてくださいね。一緒に最適化の泥沼に浸かりましょう！
コメント