是否存在基于移位值正负实现左右移位的AVX2内置指令？

注意：C标准中移位操作的右操作数不能为负数，你的标量代码x <<= value当value < 0时属于未定义行为，以下实现基于负移位值表示左移其绝对值，正移位值表示右移其绝对值的合理需求假设。

AVX2 实现方案 #

AVX2指令集没有直接的单条内置指令能完成“根据移位值正负自动选择左移/右移”的操作，你当前“分别计算左右移结果再混合”的思路是可行的，这里可以优化混合步骤的实现：

• 先对移位值取绝对值，统一用于左右移操作
• 生成掩码向量：标记出需要左移的元素（移位值为负的位置）
• 使用_mm256_blendv_epi32根据掩码混合左右移结果

示例代码：

__m256i shiftValues = ...; // 输入的32位移位值向量（每个元素为正/负整数）
__m256i x = ...; // 输入的待移位32位无符号整数向量

// 计算移位值的绝对值，统一用于左右移
__m256i abs_shift = _mm256_abs_epi32(shiftValues);

// 分别计算左移和右移结果
__m256i shifted_left = _mm256_sllv_epi32(x, abs_shift);
__m256i shifted_right = _mm256_srlv_epi32(x, abs_shift);

// 生成掩码：移位值为负的位置对应0xFFFFFFFF，其余为0x00000000
__m256i is_negative = _mm256_cmpgt_epi32(_mm256_setzero_si256(), shiftValues);
__m256i mask = _mm256_and_si256(is_negative, _mm256_set1_epi32(0xFFFFFFFF));

// 根据掩码混合结果：掩码为1时选左移结果，为0时选右移结果
__m256i result = _mm256_blendv_epi32(shifted_right, shifted_left, mask);

如果需要处理有符号整数的算术右移（保留符号位），只需将_mm256_srlv_epi32替换为_mm256_srav_epi32即可。

AVX-512 参考方案 #

AVX-512（带VL扩展，支持256位向量）没有单条指令直接实现该功能，但可以利用掩码寄存器指令大幅优化混合步骤，硬件原生支持的掩码操作延迟更低、吞吐量更高：

• 生成8位掩码寄存器（对应256位向量的8个32位元素），标记需要左移的位置
• 先执行全量右移，再用带掩码的左移指令覆盖需要左移的元素

示例代码：

__m256i shiftValues = ...;
__m256i x = ...;

__m256i abs_shift = _mm256_abs_epi32(shiftValues);
// 生成掩码：移位值为负的位置对应1，其余为0
__mmask8 neg_mask = _mm256_movemask_epi32(_mm256_cmpgt_epi32(_mm256_setzero_si256(), shiftValues));

// 先执行全量右移，再对掩码标记的元素执行左移并覆盖结果
__m256i result = _mm256_srlv_epi32(x, abs_shift);
result = _mm256_mask_sllv_epi32(result, neg_mask, x, abs_shift);

同样，若需算术右移，替换_mm256_srlv_epi32为_mm256_srav_epi32即可。

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