使用SIMD提取位

Question

使用SIMD提取位

5

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

此问题类似于使用位操作提取位，但我希望使用SIMD指令来解决，因为这样会更快。

- jianqiang

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- ErmIg · Accepted Answer

使用内置函数_mm256_permutevar8x32_epi32()选择每个元素的高3位，并使用它选取所需的32位元素。
使用内置函数_mm256_sllv_epi32()选择向量中每个元素的低5位，并创建一个位掩码。
使用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)); }