メモリ律速なプログラムを高速化するには、キャッシュ上のデータを再利用し、遅い主記憶(DRAM)へのアクセスを減らすことが重要です。
本稿では、キャッシュのレイテンシを実測する方法を解説します。 キャッシュに関する情報は CPU の仕様書にも記載されていますが、どうすれば仕様書通りの性能を出せるか知っておくことは有益ですし、未知のプロセッサが出てきた場合にも計測できると便利ですからね。
メモリ階層
DRAM 上のデータは、キャッシュライン単位1でキャッシュに読み込まれます。 読み込まれたデータはしばらくの間キャッシュに置かれているため、同じキャッシュライン上のデータに複数回アクセスする場合は、高速に取得できます。
現代の CPU の多くは、速度と容量の異なる複数のキャッシュを持っています。 一般に、容量の大きいキャッシュほどアクセスにかかる時間(レイテンシ)も大きくなります。
レジスタ、キャッシュ、DRAM の速度と容量の関係
CPU はメモリ上のデータにアクセスする際、まず L1 キャッシュに目的のデータが存在するかを確認します。 存在する場合(キャッシュヒット時)はそれを参照し、存在しない場合(キャッシュミス時)は、L2/L3 キャッシュ、DRAM からデータを取得します。
キャッシュ構成の例
例として、AMD Ryzen 7 4700U (Zen 2) のキャッシュ構成を lstopo コマンドで確認してみます。
|
|
AMD Ryzen 7 4700U のキャッシュ構成
- CPU コアは 8 個
- L1 キャッシュと L2 キャッシュはコアごとに独立しており、容量は 32 KiB と 512 KiB
- L3 キャッシュはコア 0〜3 とコア 4〜7 でそれぞれ共有されており、容量は 4 MiB × 2
各階層のキャッシュラインサイズや way 数は次のコマンドで取得できます。
|
|
結果を整理すると、次の表のようになります。 この CPU では L2 キャッシュが Inclusive なので、L1 キャッシュに存在するキャッシュラインは L2 キャッシュにも置かれていることが保証されます。
| キャッシュ | 容量 | キャッシュラインサイズ | way 数 | Inclusive |
|---|---|---|---|---|
| L1d | 32 KiB / core | 64 B | 8-way | 0 |
| L2 | 512 KiB / core | 64 B | 8-way | 1 |
| L3 | 4 MiB / 4 cores | 64 B | 16-way | 0 |
キャッシュレイテンシの計測
L1/L2/L3 キャッシュのロードレイテンシを計測するベンチマークを作成します。 完全なソースコードはこちらです。
バッファ内の要素にランダムアクセスし、ロード 1 回あたりにかかる時間を計測します。 データの供給元によってロードレイテンシがどう変わるかを確認するため、バッファサイズを L1 キャッシュより小さいものから、L3 キャッシュより十分大きいものまで変化させます。 実行時に各キャッシュ階層のキャッシュミス率を計測し、L1/L2/L3 キャッシュのレイテンシを推定します。
バッファサイズごとに 5 種類の乱数シードでベンチマークを実行し、得られたレイテンシの中央値を、そのバッファサイズのロードレイテンシとします。 各シードでは、100 万回のロードを 1 回の試行とし、3 回のウォームアップ実行後に 10 回計測して、サイクル数が最小となった試行の結果を記録します。
ハードウェアプリフェッチ対策
バッファ内の要素に対して規則的にアクセスすると、CPU はアクセスパターンから将来アクセスされる可能性の高い要素を予測し、先行してキャッシュに読み込もうとします。 この機能が有効に働くと、ロードレイテンシを正確に推定するのが難しくなってしまいます。 そこで、アクセスパターンが予測しにくくなるように工夫します。
バッファ内の要素をランダムな順序でつないだ連結リスト(pointer chain)を作り、ポインタを順に辿って要素にアクセスします。 次にアクセスする要素のアドレスは直前のロードが完了するまで確定しないため、アクセス先を事前に予測することが難しくなります。
pointer chain
また、各要素が 1 つのキャッシュラインを占有するようにパディングし、1 回のロードで複数の要素がキャッシュに読み込まれてしまうことを防ぎます。
TLB ミス対策
バッファサイズを大きくすると、アクセスするページ数が増え、TLB ミスの影響が大きくなります。 そこで、2 MiB の huge page を使用してバッファを確保することで、TLB ミスを抑制します。
|
|
これを使うためには、あらかじめ huge page を用意しておく必要があります。 512 個の huge page を確保する場合、ベンチマーク実行前に次のコマンドを実行します。
|
|
キャッシュミス率の計測
perf-counter を使ってキャッシュミス数を計測します。
AMD Ryzen 7 4700U (Zen 2) には、L1 データキャッシュへのデータ供給回数を供給元ごとに数えるイベント DATA_CACHE_REFILLS_FROM_SYSTEM が用意されています。
|
|
MABRESP_LCL_L2 は同一コアの L2 キャッシュからの供給、LS_MABRESP_LCL_CACHE と LS_MABRESP_RMT_CACHE はそれ以外のキャッシュからの供給、LS_MABRESP_LCL_DRAM と LS_MABRESP_RMT_DRAM は DRAM からの供給を表します。
これらを組み合わせて、L1/L2/L3 キャッシュミス数を計測するイベントを次のように定義します。
|
|
L1 データキャッシュミス数は、L1 データキャッシュにデータが供給された回数です。 L2 キャッシュミス数は、そのうち、同一コアの L2 キャッシュから供給されたものを除いた回数です。 L3 キャッシュミス数は、DRAM からデータが供給された回数を表します。 各カウントをロード回数で割り、L1/L2/L3 キャッシュミス率を求めます。
キャッシュレイテンシの推定
計測で得られた L1/L2/L3 キャッシュミス率をそれぞれ とすると、データの供給元(L1、L2、L3、DRAM)の割合 は、
と表せます。
L1、L2、L3、DRAM からデータを取得する際のレイテンシをそれぞれ とすると、実測レイテンシ は次式で表せます。
この式を基に、計測結果から各キャッシュのレイテンシを推定します。
まず、L1 ミス率が十分に小さい範囲では、すべてのロードが L1 でヒットするとみなし、実測値 を L1 レイテンシ とします。
次に、L1 ミス率が閾値以上で、L2 ミス率が十分に小さい範囲では、すべてのロードが L1 または L2 でヒットするとみなし、既に求めた を使って を推定します。
同様に、L2 ミス率が閾値以上で、L3 ミス率が閾値未満の範囲では、ロードが L1、L2、L3 のいずれかでヒットするとみなし、 を推定します。
今回は、ミス率の閾値を 0.01% とします。 条件を満たすバッファサイズが複数ある場合は、それぞれから得られた推定値の中央値を、そのキャッシュのレイテンシとして採用します。
ビルドと実行
リポジトリのルートで次のコマンドを実行し、Zen 2 向けにビルドします。 使用した GCC のバージョンは 15.2.1 です。
|
|
CPU コア 7 番を指定し、プロセスの優先度を 99 にしてベンチマークを実行します。
|
|
計測結果
AMD Ryzen 7 4700U (Zen 2) での計測結果を次の図に示します2。
AMD Ryzen 7 4700U の計測結果。上図は、バッファサイズに対する実測レイテンシと L1/L2/L3 キャッシュの推定レイテンシを示す。下図は、ロードが L1/L2/L3 キャッシュでヒットした割合、および DRAM から取得した割合を示す。
バッファサイズが L1 キャッシュ容量(32 KiB)以下の場合、L1 キャッシュヒット率はほぼ 100% です。 32 KiB を超えると、L2 キャッシュからデータを取得する割合が上がり、ロードレイテンシが増加します。 バッファサイズをさらに大きくすると、L2、L3 キャッシュについても同様に、各キャッシュの容量を超えた付近で次の階層からデータを取得する割合が上がり、ロードレイテンシが増加します。
バッファサイズが L3 キャッシュの容量よりも十分大きくなると、ほとんどのデータを DRAM から取得するようになり、ロードレイテンシは約 225 サイクルまで増加します。
各キャッシュ階層の推定レイテンシは次の表の通りです。 いずれも AMD のドキュメントに記載されている値とおおむね一致しています。
| キャッシュ | ロードレイテンシの推定値 | AMD のドキュメントに記載されている値 |
|---|---|---|
| L1d | 4.0 サイクル | 4〜5 サイクル |
| L2 | 12.0 サイクル | 12 サイクル以上 |
| L3 | 38.6 サイクル | 平均 39 サイクル |
参考
-
64 バイトであることが多いです。 ↩︎
-
結果のプロットには
analyze_cache_latency.pyを使用しました。 ↩︎