performance - _mm_shuffle_epi8 内在函数的用法

Question

有人可以解释一下_mm_shuffle_epi8SSSE3内在吗？我知道它会随机播放 16 个 8 位整数，__m128i但不确定如何使用它。

我基本上想用_mm_shuffle_epi8修改下面的函数来获得更好的性能。

while(not done)
    dest[i+0] = (src+j).a;
    dest[i+1] = (src+j).b;
    dest[i+2] = (src+j).c;
    dest[i+3] = (src+j+1).a;
    dest[i+4] = (src+j+1).b;
    dest[i+5] = (src+j+1).c;
    i+=6;
    j+=2;

Answer 1

_mm_shuffle_epi8（更好地称为pshufb），基本上是这样做的：

temp = dst;
for (int i = 0; i < 16; i++)
    dst[i] = (src[i] & 0x80) == 0 ? temp[src[i] & 15] : 0;

至于这里能不能用，不知道涉及的类型就无法判断了。无论如何它都不会“很好”，因为目标是 6 个字节的块（或单词？或 dwords？）。您可以通过展开并进行大量移位和或操作来完成这项工作。

Answer 2

这是使用内在函数的示例；您必须了解如何将其应用于您的特定情况。此代码一次交换 4 个 32 位整数：

unsigned int *bswap(unsigned int *destination, unsigned int *source, int length) {
    int i;
    __m128i mask = _mm_set_epi8(12, 13, 14, 15, 8, 9, 10, 11, 4, 5, 6, 7, 0, 1, 2, 3);
    for (i = 0; i < length; i += 4) {
        _mm_storeu_si128((__m128i *)&destination[i],
        _mm_shuffle_epi8(_mm_loadu_si128((__m128i *)&source[i]), mask));
    }
    return destination;
}