Intel Sapphire Rapids は AVX512_FP16 をサポート

モバイル向けでは Ice Lake (client) から、デスクトップ向けでは 2021年第一四半期での登場が予告されている Rocket Lake からついにサポートされる AVX512命令。
これまでに何度も拡張を繰り返して来たが、Sapphire Rapids ではさらに拡張された AVX512_FP16 がサポートされる。

Linux Kernel に投稿されたパッチのコメントによれば、AVX512_FP16 は Sapphire Rapids 等の Intelプロセッサでサポートされ、精度や大きさを確保しつつ、FP32 よりも高速で優れた性能を得られる、としている。

AVX512_FP16 is supported by Intel processors, like Sapphire Rapids.
It could gain better performance for it’s faster compared to FP32
while meets the precision or magnitude requirement. It’s availability
is indicated by CPUID.(EAX=7,ECX=0):EDX[bit 23]

引用元:LKML: Kyung Min Park: [PATCH 2/2] x86: Expose AVX512_FP16 for supported CPUID

AVX512系命令では、AVX512_VNNI にて精度を下げて複数の要素を扱うことで処理性能を向上させることができ、データ精度 FP32 の場合に対してピーク理論性能が、 Int8 では 4倍、Int16 では 2倍となる。(ただし、メモリ性能がボトルネックとなるため、実際の性能はそれより低くなる可能性がある。)
AVX512_FP16 はこれを拡張したものと思われ、ピーク理論性能は Int16 と同じとなるが、FP16 であれば Int16 よりも数値が広い範囲で保持される。

AVX512、FP16 というキーワードで、過去にバグ報告プラットフォームである intel in Launchpad にちらと出てきた HFNI 命令が思い出される。
Intel Alder Lake、Sapphire Rapids にて追加される2つの命令 ――AVX2 VNNI /HFNI | Coelacanth’s Dream その時のリンク先は消されてしまったが、メールアーカイブには残っていた。¹
HFNI は AVX512命令において、FP32 の代わりに FP16 を使用し、性能を最大 2倍に引き上げるものと説明されている。
SPR (Sapphire Rapids) で新しくサポートされる命令としているが、タイトルには ADL (Alder Lake) があることから、Sapphire Rapids と言うより Golden Cove アーキテクチャ で新たにサポートする命令と考えられる。
用途として 5G基地局等が想定されており、Intel が過去に CPUマイクロアーキテクチャのロードマップを発表した時、Golden Cove の特徴に Network/5G Perf があったことも上と一致する。

AVX512_FP16 と HFNI の関係については、HFNI が未だに Intel の 一般にも公開されている限りだが 公式ドキュメントに現れていないことから (AVX512_FP16 はある)、違うのは名前だけで、命令の中身は一緒ではないかと思われる。
かなり似ていながら僅かに違いが存在するとかだと、ただでさえややこしい AVX512命令群がさらにややこしくなってしまう。

Intel AVX サポート表

CPUマイクロアーキテクチャの違いと併せて、対応範囲が混沌としている AVX512命令群だが、GCC 等コンパイラのドキュメントを頼りに頑張って整理したのが下の表となる。
プロセッサのコードネームを略称で記載しているが、一応書いておくと、SKX (Skylake server) 、CNL (Cannon Lake) 、CLX (Cascade Lake) 、CPX (Cooper Lake) 、ICL (Ice Lake client) 、ICX (Ice Lake server) 、TGL (Tiger Lake) 、SPR (Sapphire Rapids) 、ADL (Alder Lake) と対応している。
また、Alder Lake に関しては、Intel® Architecture Instruction Set Extensions Programming Reference に 「Intel Hybrid Technology は AVX512 をサポートしない」とあり、そのためアーキテクチャ自体がサポートしていない場合と区別してある。

AVX512命令は Xeon Phi シリーズ 、Knight’s Landing 、Knights Mill でも一部サポートされているが、それも含めるとさらに表が肥大化してしまうため省いた。5つの命令をまとめている 2行目を細かく分ける必要が出てしまう。
それと Rocket Lake も省いているが、その Crypress Cove アーキテクチャ は Ice Lake (client) と同じ機能を備えることから、AVX512命令の範囲も同様と思われる。