CPU優(yōu)化技術之NEON 的基本原理、指令

本文選自極術專欄《嵌入式AI》的文章，授權轉自作者高性能計算學院的《移動端算法優(yōu)化》。前面我們學習了如何快速上手開始NEON編程，ArmNEON優(yōu)化技術以及Arm NEON學習（三）NEON 匯編與Intrinsics編程。本篇將從CPU優(yōu)化技術層面來詳細講解Arm NEON。 ? ?

一、SIMD

Arm NEON 是適用于 Arm Cortex-A 和 Cortex-R 系列處理器的一種 SIMD（Single Instruction Multiple Data）擴展架構。

SIMD 采用一個控制器來控制多個處理器，同時對一組數(shù)據(jù)（又稱“數(shù)據(jù)向量”）中的每個數(shù)據(jù)分別執(zhí)行相同操作，從而實現(xiàn)并行技術。

SIMD 特別適用于一些常見的任務，如音頻圖像處理。大部分現(xiàn)代 CPU 設計都包含了 SIMD 指令，來提高多媒體使用的性能。

SIMD 操作示意圖

如上圖所示，標量運算時一次只能對一對數(shù)據(jù)執(zhí)行乘法操作，而采用 SIMD 乘法指令，則一次可以對四對數(shù)據(jù)同時執(zhí)行乘法操作。

A. 指令流與數(shù)據(jù)流

費林分類法根據(jù)指令流（Instruction）和數(shù)據(jù)流（Data）的處理方式進行分類，可分成四種計算機類型：

費林分類示意圖

1. SISD（Single Instruction Single Data）

機器的硬件不支持任何形式的并行計算，所有的指令都是串行執(zhí)行。單個核心執(zhí)行單個指令流 , 操作存儲在單個內(nèi)存中的數(shù)據(jù) , 每次一個操作。早期的計算機都是SISD機器，如馮諾.依曼架構，IBM PC機等。

2. MISD（Multiple Instruction Single Data）

是采用多個指令流來處理單個數(shù)據(jù)流。由于實際情況中，采用多指令流處理多數(shù)據(jù)流才是更有效的方法，因此MISD只是作為理論模型出現(xiàn)，沒有投入到實際應用之中。

3. MIMD（Mutiple Instruction Mutiple Data）

計算機具有多個異步和獨立工作的處理器。在任何時鐘周期內(nèi)，不同的處理器可以在不同的數(shù)據(jù)片段上執(zhí)行不同的指令，也即是同時執(zhí)行多個指令流，而這些指令流分別對不同數(shù)據(jù)流進行操作。MIMD架構可以用于諸如計算機輔助設計、計算機輔助制造、仿真、建模、通信交換機的多個應用領域。

除了以上模型外，由NVIDIA公司生產(chǎn)的GPU引入SIMT體系結構：

4. SIMT（Single Instruction Multiple Threads）

類似 CPU 上的多線程，所有的核心各有各的執(zhí)行單元，數(shù)據(jù)不同，執(zhí)行的命令是相同的。多個線程各有各的處理單元，和 SIMD 共用一個 ALU 不同。

SIMT 示意圖

B. SIMD 特點及發(fā)展趨勢

1. SIMD 優(yōu)勢與不足

2. SIMD發(fā)展趨勢

以Arm架構下的下一代 SIMD 指令集?SVE（Scalable Vector Extension，可擴展矢量指令）為例，其是_針對高性能計算（HPC）和機器學習等領域開發(fā)的一套全新的矢量指令集_。

SVE 指令集中有很多概念與 NEON 指令集類似，例如矢量、通道、數(shù)據(jù)元素等。

SVE指令集也提出了一個全新的概念：可變矢量長度編程模型。

SVE 可擴展模型

傳統(tǒng)的 SIMD 指令集采用固定大小的向量寄存器，例如 NEON 指令集采用固定的 64/128 位長度的矢量寄存器。

而支持 VLA 編程模型的 SVE 指令集則支持可變長度的矢量寄存器。因此允許芯片設計者根據(jù)負載和成本來選擇一個合適的矢量長度。

SVE 指令集的矢量寄存器的長度最小支持 128 位，最大可以支持 2048 位，以 128 位為增量。SVE 設計確保同一個應用程序可以在支持不同矢量長度的 SVE 指令機器上運行，而不需要重新編譯代碼。

Arm 在 2019 年便推出了 SVE2，以最新的 Armv9 為基礎，擴充了更多的運算類型以全面替代 NEON，同時增加了矩陣相關運算的支持。

二、 Arm?的 SIMD 指令集

1. Arm?處理器的 SIMD 支持 - NEON

Arm NEON 單元默認包含在 Cortex-A7 和 Cortex-A15 處理器中，但在其他 Armv7 Cortex-A 系列處理器中是可選的，某些實現(xiàn) Armv7–A 或 Armv7–R 架構配置文件的Cortex-A 系列處理器可能不包含NEON單元。

符合 Armv7 的內(nèi)核的可能組合有以下四種：

因此必須首先確認處理器是否支持 NEON 和 VFP?？梢栽诰幾g和運行的時候進行檢查。

NEON 發(fā)展史

2. ARM 處理器的 SIMD 支持檢查

2.1 編譯階段檢查

檢測 NEON 單元是否存在的最簡單方法。在 Arm 編譯器工具鏈（armcc）v4.0 及更高版本或 GCC 中，檢查預定義宏?ARM_NEON?或者?__arm_neon?是否開啟。

armasm?等效的預定義宏是?TARGET_FEATURE_NEON。

2.2 運行階段檢查

在運行時檢測 NEON 單元需要操作系統(tǒng)的幫助。ARM 架構有意不向用戶模式應用程序公開處理器功能。在Linux下，/proc/cpuinfo?以可讀的形式包含此信息，比如：

在Tegra（帶有FPU的雙核Cortex-A9處理器）

$ /proc/cpuinfo 
swp half thumb fastmult vfp edsp thumbee vfpv3 vfpv3d16

帶有 NEON 單元的 ARM Cortex-A9 處理器

$ /proc/cpuinfo 
swp half thumb fastmult vfp edsp thumbee neon vfpv3

由于?/proc/cpuinfo?輸出是基于文本的，因此通常首選查看輔助向量?/proc/self/auxv，其包含二進制格式的內(nèi)核?hwcap，可以輕松地在?/proc/self/auxv?文件中搜索?AT_HWCAP?記錄，以檢查?HWCAP_NEON?位（4096）。

某些 Linux 發(fā)行版?ld.so?鏈接器腳本被修改為通過 glibc 讀取?hwcap?，并為啟用 NEON 的共享庫添加額外的搜索路徑。

3. 指令集關系

在Armv7中，NEON 與 VFP 指令集具有以下關系：

具有 NEON 單元但沒有VFP單元的處理器無法在硬件中執(zhí)行浮點運算。

由于 NEON SIMD 操作更有效地執(zhí)行向量計算，因此從 ARMv7 的引入開始，VFP 單元中的向量模式操作已被棄用。因此，VFP 單元有時也稱為浮點單元（FPU）。

VFP 可以提供完全兼容 IEEE-754 的浮點運算，Armv7 NEON 單元中的單精度運算不完全符合 IEEE-754。

NEON不能取代 VFP。VFP 提供了一些在 NEON 指令集中沒有等效實現(xiàn)的專用指令。

半精度指令僅適用于包含半精度擴展的 NEON 和 VFP 系統(tǒng)。

在Armv8中，VFP已被NEON取代，以上問題如 NEON 并不完全符合 IEEE 754 標準，并且有一些指令 VFP 支持而 NEON 不支持的問題已在 ARMv8 中得到解決。

三、NEON

NEON 是適用于 Arm Cortex-A 系列處理器的一種128位 SIMD 擴展結構，每個處理器核心均有一個 NEON 單元，因此可以實現(xiàn)多線程并行的加速效果。

1. NEON基本原理

1.1 NEON 指令執(zhí)行流程

上圖為 NEON 單元完成加速計算的流程圖。其中向量寄存器中的每個元素同步執(zhí)行計算，以此來加速計算過程。

1.2 NEON 計算資源

NEON 與 Arm?處理器資源關系

- NEON 單元作為 Arm指令集的擴展，使用獨立于 ARM 原有寄存器的 64位 或 128 位寄存器進行 SIMD 處理，在 64位 寄存器的寄存器文件上運行。
- NEON 和 VFP 單元完全集成到了處理器中，并共享處理器資源以進行整數(shù)運算、循環(huán)控制和緩存。
與硬件加速器相比，這顯著降低了面積和功耗成本。并且其還使用更簡單的編程模型，因為NEON 單元使用與應用程序相同的地址空間。

NEON 與 VFP 資源關系

NEON 寄存器與 VFP 寄存器重疊，Armv7 有 32 個 NEON D 寄存器，如下圖所示。

NEON 寄存器

2. NEON指令

2.1 自動矢量化

向量化編譯器可以使用 C 或 C++ 源代碼，以一種能夠有效使用 NEO N硬件的方式對其進行矢量化。這意味著可以通過編寫可移植的 C 代碼，同時仍然可以獲得 NEON 指令所帶來的性能水平。

為了幫助矢量化，將循環(huán)迭代次數(shù)設為矢量長度的倍數(shù)。GCC 和 ARM 編譯器工具鏈都具有為 NEON 技術啟用自動矢量化的選項。

2.2 NEON匯編

對于性能要求特別高的程序，手工編寫匯編代碼是更適合的方式。

GNU 匯編器（gas） 和 Arm Compile r工具鏈匯編器（armasm）都支持 NEON 指令的匯編。

編寫匯編函數(shù)時，需要了解?Arm?EABI，其定義了如何使用寄存器。ARM嵌入式應用程序二進制接口（EABI）指定哪些寄存器用于傳遞參數(shù)、返回結果或必須保留，指定了除Arm內(nèi)核寄存器之外的32個D寄存器的使用。下圖對寄存器功能進行了總結。

寄存器功能

2.3 NEON Intrinsics

NEON intrinsic 函數(shù)提供了一種編寫 NEON 代碼的方法，該方法比匯編代碼更易于維護，同時仍然可以控制生成的 NEON 指令。

內(nèi)部函數(shù)使用與 D 和 Q NEON 寄存器對應的新數(shù)據(jù)類型。數(shù)據(jù)類型支持創(chuàng)建直接映射到NEON 寄存器的 C 變量。

NEON intrinsic 函數(shù)的編寫類似于使用這些變量作為參數(shù)或返回值的函數(shù)調用。編譯器做了一些通常與編寫匯編語言相關的繁重工作，例如：

寄存器分配
代碼調度或重新排序指令

intrinsic 缺點

無法讓編譯器準確輸出想要的代碼，因此在轉向NEON匯編代碼時仍有一些改進的可能性。

NEON 指令簡類型

NEON 數(shù)據(jù)處理指令可以分為正常指令、長指令、寬指令、窄指令和飽和指令。
以 Intrinsic 的長指令為例?int16x8_t vaddl_s8(int8x8_t __a, int8x8_t __b);
- 上面的函數(shù)將兩個64位的 D 寄存器向量（每個向量包含8個8位數(shù)字）相加，生成一個包含8個16位數(shù)字的向量（存儲在128位的Q寄存器中），從而避免相加的結果溢出。

四、其他 SIMD 技術

1. 其他平臺上的 SIMD 技術

SIMD 處理不是 Arm 獨有的，下圖將其與 x86 和 Altivec 進行了比較。

SIMD 對比

2. 與專用 DSP 對比

許多基于 Arm 的 SOC 中還包含 DSP 等協(xié)處理硬件，因此可以同時包含 NEON 單元和DSP。相對于 DSP，NEON 的特點有：

五、總結

本節(jié)主要介紹基本 SIMD 及其他的指令流與數(shù)據(jù)流的處理方式，NEON 的基本原理、指令以及與其他平臺及硬件的對比。

期望大家都能有所收獲。

編輯：黃飛

?