基于AVX512f的16个u32对齐块连续递增SIMD检测问询

针对16个u32的对齐块，利用AVX512指令集可高效完成连续递增序列的检测，以下是两种可行算法的实现思路及Rust代码（基于packed_simd crate）：

算法A：常量偏移比较法 #

思路 #

1. 预定义常量向量DECREASE_U32，元素为[15, 14, 13, ..., 0]，对应每个位置的偏移量
2. 输入向量与该常量向量相加（允许无符号溢出），若输入是连续递增序列，相加后所有元素会等于同一个值（即第一个元素+15）
3. 将相加结果与该值的广播向量比较，若所有元素相等则判定为连续递增序列

Rust代码实现 #

use packed_simd::u32x16;

// 预定义常量：[15,14,...,0]
const DECREASE_U32: u32x16 = u32x16::new(15,14,13,12,11,10,9,8,7,6,5,4,3,2,1,0);

fn is_consecutive_increasing_a(input: u32x16) -> bool {
    let a = input.wrapping_add(DECREASE_U32);
    let first_val = a.extract(0);
    let compare_vec = u32x16::splat(first_val);
    // 检查所有元素是否相等
    a.eq(compare_vec).all()
}

算法B：相邻元素差检测法 #

思路 #

1. 将输入向量向右移位一位（即第二个元素移到第一个位置，第三个到第二个，以此类推，最后一个位置补任意值）
2. 原输入向量减去移位后的向量，若输入是连续递增序列，除第一个元素外，其余元素的差值均为1
3. 生成掩码排除第一个元素，检查剩余元素是否全为1

Rust代码实现 #

use packed_simd::u32x16;

fn is_consecutive_increasing_b(input: u32x16) -> bool {
    // 将输入向量向右移位一位：[input[1], input[2], ..., input[15], 0]
    let shifted = input.shift_right(1);
    let diff = input.wrapping_sub(shifted);
    // 生成掩码：第一个元素为false，其余为true
    let mask = u32x16::new(false, true, true, true, true, true, true, true,
                           true, true, true, true, true, true, true, true);
    // 检查掩码覆盖的元素是否全为1
    diff.eq(u32x16::splat(1)).and(mask).all()
}

补充说明 #

• 算法A的指令开销更低，仅包含加法、广播和全相等检查，适合对性能要求较高的场景
• 算法B需要移位和掩码操作，但逻辑更直观，适合需要明确验证相邻元素差的场景

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