用SIMD提取比特

Question

我想从一个由8个整数组成的另一个__mm256i offset指定的8个位置从一个寄存器变量__mm256i src中提取8个比特。 例如：如果offset是[1,3,5,21,100,200,201,202]，我想从src获得第1,3,5,10,100,121,202位，并将它们打包到int8。

这个问题类似于Extracting bits using bit manipulation，但我想用SIMD指令的解决方案，因为它的速度要快得多。

Answer 1

1. 在每个元素中选择高位3位，并使用固有的_mm256_permutevar8x32_epi32（）选择所需的32位元素。
2. 在矢量的各元素中选择低5位，并与使用的固有_mm256_sllv_epi32创建位掩码（）。
3. 包结果与使用的_mm256_movemask_ps（）INT8（投__m256i到__m256）。

下面有一个例子：

uint8_t Select(__m256i offset, __m256i src) 
{ 
    __m256i permutedSrc = _mm256_permutevar8x32_epi32(src, _mm256_srli_epi32(offset, 5)); 
    __m256i shift = _mm256_and_si256(offset, _mm256_set1_epi32(31)); 
    __m256i bitmask = _mm256_sllv_epi32(_mm256_set1_epi32(1), shift); 
    __m256i mask = _mm256_cmpeq_epi32(_mm256_and_si256(permutedSrc, bitmask), _mm256_setzero_si256()); 
    return ~_mm256_movemask_ps(_mm256_castsi256_ps(mask)); 
}