如何在ARMv7平台通过NEON高效实现向量(SIMD)就近舍入偶数取整？

针对ARMv7 NEON无法直接使用ARMv8的vcvtnq_s32_f32等就近舍入偶数取整指令的问题，以下提供两种纯向量实现方案，均避免了标量循环的低效操作：

方案一：基于浮点偏移的高效实现（适用于绝对值<2^23的浮点数） #

利用单精度浮点数的尾数特性，通过浮点加法的自动舍入实现就近偶数取整（ARMv7 NEON浮点运算默认舍入模式为就近偶数）：

#include <arm_neon.h>

int32x4_t vcvtnq_s32_f32_armv7(float32x4_t x) {
    const float32x4_t OFFSET = vdupq_n_f32(8388608.0f); // 2^23
    const float32x4_t ABS_X = vabsq_f32(x);
    // 生成掩码：判断数值是否超出[-2^23, 2^23]范围
    uint32x4_t out_of_range_mask = vcgtq_f32(ABS_X, OFFSET);

    // 范围内数值：通过偏移实现就近偶数舍入
    float32x4_t temp = vaddq_f32(x, OFFSET);
    temp = vsubq_f32(temp, OFFSET);
    int32x4_t rounded = vcvtq_s32_f32(temp);

    // 超出范围数值：单精度浮点数已无法表示小数部分，直接向零舍入即可
    int32x4_t direct_conv = vcvtq_s32_f32(x);

    // 根据掩码选择对应结果
    return vbslq_s32(out_of_range_mask, direct_conv, rounded);
}

原理说明 #

• 单精度浮点数的尾数为23位，当数值绝对值小于2^{23时，加上2}23会触发浮点加法的就近偶数舍入，自动将数值对齐到最近整数；减去2^23后即可得到舍入后的整数值，再转成int32类型。
• 当数值绝对值≥2^23时，单精度浮点数已无法表示小数部分（此时数值本身就是整数），直接使用vcvtq_s32_f32向零舍入即可得到正确结果。

方案二：基于规则判断的全范围实现 #

直接遵循就近偶数取整（银行家舍入）的规则，通过NEON向量指令完成条件判断与调整，适用于所有单精度浮点数范围：

#include <arm_neon.h>

int32x4_t vcvtnq_s32_f32_armv7_full(float32x4_t x) {
    // 1. 向零舍入得到整数部分
    int32x4_t int_x = vcvtq_s32_f32(x);
    // 2. 计算小数部分的绝对值
    float32x4_t frac = vsubq_f32(x, vcvtq_f32_s32(int_x));
    frac = vabsq_f32(frac);

    // 3. 生成调整掩码
    // 掩码1：小数部分>0.5，需要加1
    uint32x4_t mask_gt_half = vcgtq_f32(frac, vdupq_n_f32(0.5f));
    // 掩码2：小数部分=0.5且整数部分为奇数，需要加1
    uint32x4_t mask_eq_half = vceq_f32(frac, vdupq_n_f32(0.5f));
    uint32x4_t int_x_odd = vtstq_s32(int_x, vdupq_n_s32(1));
    uint32x4_t mask_tie_odd = vandq_u32(mask_eq_half, int_x_odd);

    // 合并需要加1的掩码
    uint32x4_t mask_add_one = vorrq_u32(mask_gt_half, mask_tie_odd);

    // 生成要添加的数值：符合条件加1，否则加0
    int32x4_t add = vbslq_s32(mask_add_one, vdupq_n_s32(1), vdupq_n_s32(0));

    // 4. 得到最终结果
    return vaddq_s32(int_x, add);
}

原理说明 #

就近偶数取整的核心规则：

• 小数部分<0.5：保留整数部分
• 小数部分>0.5：整数部分加1
• 小数部分=0.5：取最近的偶数（整数部分为奇数则加1，偶数则保持）

通过NEON向量指令批量完成所有元素的条件判断与调整，全程无标量操作，效率远高于逐个元素处理的方案。

关于之前错误方案的说明 #

你之前实现的round2even函数逻辑为(x*0.5 + 0.5)转整数后左移1，这是舍入到最近的偶数（如1.1→2），而非标准的就近舍入偶数取整（1.1应舍入到1），因此不符合需求。

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