如何将__m256d比较掩码转换为适配_mm_maskstore_epi32的__m128i掩码

错误根因 #

你的原有写法存在两点核心问题：

1. _mm256_castpd256_pd128仅提取256位掩码的低128位，直接丢失了第3、4个双精度浮点数的比较结果
2. 低128位包含两个64位的全1/全0掩码，直接转成128位整数向量后会被拆分为4个32位值，导致前两个32位对应第一个双精度的比较结果、后两个32位对应第二个双精度的比较结果，和4个32位整数的一一对应逻辑完全不符

转换规则说明 #

先明确两类掩码的要求：

• _mm256_cmp_pd生成的__m256d掩码，每个64位 lane 全为1表示匹配，全为0表示不匹配
• _mm_maskstore_epi32要求的__m128i掩码，每个32位 lane 仅最高位为1时才会写入对应内存

正确实现 #

AVX2环境（性能最优） #

直接用带符号整数截断指令完成转换，没有额外开销：

// 将__m256d比较掩码转为256位整数向量，和你之前双精度场景的转换逻辑一致
__m256i mask_64 = _mm256_castpd_si256(cmp_pd_mask);
// 4个64位整数截断为4个32位整数，打包为128位向量
// 全1的64位值截断为32位仍是全1（最高位为1），全0截断后仍是全0，完全符合掩码要求
__m128i mask_32 = _mm256_cvtepi64_epi32(mask_64);
// 直接传入maskstore接口使用
_mm_maskstore_epi32(int_output_ptr, mask_32, int_vector);

纯AVX兼容实现 #

如果运行环境不支持AVX2，可以通过位提取构造目标掩码：

// 提取4位掩码，每一位对应一个双精度的比较结果
int mask_bits = _mm256_movemask_pd(cmp_pd_mask);
// 构造每个32位最高位对应比较结果的128位掩码
__m128i mask_32 = _mm_setr_epi32(
    (mask_bits & 1) ? 0x80000000 : 0,
    (mask_bits & 2) ? 0x80000000 : 0,
    (mask_bits & 4) ? 0x80000000 : 0,
    (mask_bits & 8) ? 0x80000000 : 0
);
_mm_maskstore_epi32(int_output_ptr, mask_32, int_vector);

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