c++ - 有没有办法用 AVX2 编写 _mm256_shldi_epi8(a,b,1) ？ ( vector 之间每 8 位元素移动一位)

Question

我需要将 b 的每个元素的最高位移至 a 相应元素的底部，例如 AVX512VBMI2 _mm256_shldi_epi16/32/64计数为 1。

有人知道如何改变这种方式吗？

示例:

__m256i x = { 11001100, 00110011, 11001100, 00110011,... x16 }
__m256i y = { 10111100, 10001011, 11000010, 01100111,... x16 }
__m256i res = _mm256_shldi_epi16(x,y);

然后 res 包含:

10011001、01100111、10011001、01100110、...x16

(编者注:之前的问题将其描述为 _mm256_sllv_epi8。sllv 是一个可变计数移位，其中每个元素的计数来自另一个元素中的相应元素源，并且与双类完全不同。)

Answer 1

显然，任务是将a的字节左移1，同时从b中的相应字节移入最高位，就像一个带有a的微小漏斗移位固定距离为 1。左移可以通过字节加法来完成，然后从 b 复制该位:

__m256i funnel_left1_epi8(__m256i a, __m256i b) {
    __m256i a2 = _mm256_add_epi8(a, a);
    __m256i bit_from_b = _mm256_and_si256(_mm256_srli_epi16(b, 7), _mm256_set1_epi8(1));
    return _mm256_or_si256(a2, bit_from_b);
}