例如，对于输入ymm向量x和位索引i，我希望输出向量只保留第i位，其他都归零.
使用AVX512 k寄存器，我可以编写以下内容，但AVX2及以下没有k个寄存器，那么您认为最好的方法是什么？
__m512i m512i_maskBit(__m512i x, unsigned i) {
__mmask8 m = _cvtu32_mask8(1u << i / 64);
__m512i vm = _mm512_maskz_set1_epi64(m, 1ull << i % 64);
return _mm512_and_si512(x, vm);
}
下面是一种使用可变移位的方法(仅创建掩码):
__m256i create_mask(unsigned i) {
__m256i ii = _mm256_set1_epi32(i);
ii = _mm256_sub_epi32(ii,_mm256_setr_epi32(0,32,64,96,128,160,192,224));
__m256i mask = _mm256_sllv_epi32(_mm256_set1_epi32(1), ii);
return mask;
}
_mm256_sllv_epi32(vpsllvd)由AVX2引入，它将每个32位元素移位一个可变的位数.如果(无符号)移位量大于31(即，也适用于有符号负数)，则相应的结果为0.
带小测试代码的拉杆:https://godbolt.org/z/a5xfqTcGs