ARM Neon Intrinsics中FMA指令lane参数含义及相关疑问

1. lane参数的真实含义 #

你之前对lane“按位划分打包变量”的理解，是针对普通Neon打包数据的通用lane定义，但vfmaq_laneq_f32中的lane是源向量的元素索引。

先明确指令原型：float32x4_t vfmaq_laneq_f32(float32x4_t acc, float32x4_t vec_a, float32x4_t vec_b, const int lane)

• 其中vec_b是一个包含4个单精度浮点数的128位向量（float32x4_t）
• lane参数取值范围0~3，对应vec_b中第0到第3个元素的位置
• 指令逻辑：将vec_a的每个元素与vec_b[lane]相乘，再将结果逐个累加到acc的对应元素上

简单说，这个指令是固定取vec_b中的某一个元素，和vec_a做广播乘加，而非用vec_b的所有元素做对应位置乘加。

2. 参数修改后的结果 #

全部设为0 #

每次调用都固定取vec_b的第0个元素，相当于把vec_a的所有元素都乘以vec_b[0]后累加到累加器acc。如果是矩阵乘法场景，这会导致只使用第二个矩阵的第一列参与计算，最终得到的结果矩阵所有列都会是原矩阵第一列的计算结果，完全不符合矩阵乘法的逻辑。

反转lane参数顺序（如0、1、2、3改为3、2、1、0） #

在矩阵乘法的常规实现中，lane顺序对应第二个矩阵的列索引。反转后，相当于把第二个矩阵的列顺序完全颠倒，最终得到的结果矩阵的列会是原正确结果的逆序。比如原本结果第0列对应原矩阵第0列的计算，反转后会对应原矩阵第3列的计算，以此类推。

内容的提问来源于stack exchange，提问作者Jacob FW