NEON编程中如何实现向量与标量值的乘法运算

ARM NEON 提供了直接支持向量与标量乘法的专用指令，不需要手动将标量填充为同维度向量再做运算，和你使用的fadd同属基础算术运算指令族。

对应指令说明 #

• 32位浮点场景下直接使用fmul指令即可，当其中一个源操作数为标量索引格式（如v1.s[0]，表示取v1寄存器第0个32位浮点元素作为标量），硬件会自动将该标量与向量寄存器内的每一个元素做乘法，运算效果正好匹配你给出的|3 2| *5 = |15 10|的需求。
• 指令格式示例：fmul v0.4s, v0.4s, v1.s[0]，表示将v0中4个32位浮点数分别和v1寄存器s0位置的标量相乘，结果写回v0。

参考实现代码 #

和你现有向量加法的代码风格对齐，向量乘标量的实现如下：

void vec_scalar_mul(float* dst, float* src, float scalar, int count)
{
    asm volatile (
    // 将传入的标量值加载到v1寄存器的0号32位浮点槽位
    "fmov s1, %w[scalar] \n"
    "1: \n"
    "ld1 {v0.4s}, [%[src]], #16 \n"
    // 向量逐元素乘标量
    "fmul v0.4s, v0.4s, v1.s[0] \n"
    "subs %[count], %[count], #4 \n"
    "st1 {v0.4s}, [%[dst]], #16 \n"
    "bgt 1b \n"
    : [dst] "+r" (dst), [src] "+r" (src), [count] "+r" (count)
    : [scalar] "r" (scalar)
    : "memory", "v0", "v1"
    );
}

补充说明 #

• 其他数据类型的标量乘法用法一致，只需要修改寄存器后缀即可：比如处理8位整数用.8b配b[0]、16位整数用.4h配h[0]、64位浮点数用.2d配d[0]。
• 你提供的向量加法代码存在小逻辑隐患：subs指令放在st1之前，当count不是4的整数倍时，最后一次循环会出现内存越界读写，实际生产使用时需要补充循环前对齐、循环后处理剩余不足向量长度元素的逻辑。

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