如何使用更简洁的AVX2/SIMD方法实现256位寄存器中单比特的获取与置1操作？

你说得没错，原来的实现里的switch分支和大量标量代码确实不够优雅，我们可以利用AVX2的指令特性来简化这两个函数的实现，完全摆脱对分支的依赖：

优化后的置位函数（bitset） #

__m256i bitset(__m256i source, uint8_t index) {
    // 用位运算替代除法/取模，效率更高
    const uint8_t bit_pos = index & 0x3F;  // 等价于 index % 64
    const uint8_t block_idx = index >> 6;  // 等价于 index / 64

    // 生成64位位掩码并广播到整个256位寄存器
    __m256i bitmask = _mm256_set1_epi64x(1ULL << bit_pos);
    // 将掩码移位到对应的64位块位置
    bitmask = _mm256_slli_epi64(bitmask, block_idx * 64);

    // 执行OR操作完成置位
    return _mm256_or_si256(source, bitmask);
}

优化后的位获取函数（bitget） #

这里提供两种实现思路，你可以根据场景选择：

方式一：兼顾简洁与可读性

bool bitget(__m256i source, uint8_t index) {
    const uint8_t bit_pos = index & 0x3F;
    const uint8_t block_idx = index >> 6;

    // 将目标64位块移动到寄存器最低位
    __m256i shifted = _mm256_srli_epi64(source, block_idx * 64);
    // 提取最低64位并检查目标位
    uint64_t low64 = _mm256_extract_epi64(shifted, 0);
    return (low64 & (1ULL << bit_pos)) != 0;
}

方式二：纯SIMD指令实现（无标量提取）

bool bitget(__m256i source, uint8_t index) {
    const uint8_t bit_pos = index & 0x3F;
    const uint8_t block_idx = index >> 6;

    // 生成对应位置的掩码并移位到目标块
    __m256i bitmask = _mm256_set1_epi64x(1ULL << bit_pos);
    bitmask = _mm256_slli_epi64(bitmask, block_idx * 64);

    // 用TEST指令判断掩码与源寄存器是否有重叠置位
    return !_mm256_testz_si256(source, bitmask);
}

核心优化思路 #

• 用高效位运算替代除法/取模操作，避免不必要的标量计算开销
• 利用AVX2的广播+移位指令组合，直接定位目标64位块，彻底移除switch分支
• 纯SIMD版本的位获取函数全程用向量指令完成判断，更适合后续批量处理的扩展场景

备注：内容来源于stack exchange，提问作者Maj mac