sse - 随机播放 16 位向量 SSE

Question

我正在研究 SSE 并且是这里的新手。我正在尝试使用 shuffle 指令来随机播放一个 16 位向量，如下所示:

输入:

1 2 3 4 5 6 7 8 

输出:

1 5 2 6 3 7 4 8

如何实现预期目标？我对这里使用的常量感到困惑，我没有看到任何 16 位 shuffle 指令。 shuffle 指令仅适用于 8 位和 32 位。

Answer 1

只要您可以假定 SSSE3，那么您就可以使用 pshufb aka _mm_shuffle_epi8 :

#include <stdio.h>
#include <tmmintrin.h>  // SSSE3

int main()
{
    __m128i v_in = _mm_setr_epi16(1, 2, 3, 4, 5, 6, 7, 8);
    __m128i v_perm = _mm_setr_epi8(0, 1, 8, 9, 2, 3, 10, 11, 4, 5, 12, 13, 6, 7, 14, 15);
    __m128i v_out = _mm_shuffle_epi8(v_in, v_perm); // pshufb
    printf("v_in  = %vhd\n", v_in);
    printf("v_out = %vhd\n", v_out);
    return 0;
}

编译运行:

$ gcc -Wall -mssse3 green_goblin.c 
$ ./a.out 
v_in  = 1 2 3 4 5 6 7 8
v_out = 1 5 2 6 3 7 4 8

---

仅依赖于 SSE2 的替代解决方案:

#include <stdio.h>
#include <emmintrin.h>  // SSE2

int main()
{
    __m128i v_in = _mm_setr_epi16(1, 2, 3, 4, 5, 6, 7, 8);
    __m128i v_out = _mm_shuffle_epi32(v_in, _MM_SHUFFLE(3, 1, 2, 0)); // pshufd
    v_out = _mm_shufflelo_epi16(v_out, _MM_SHUFFLE(3, 1, 2, 0));      // pshuflw
    v_out = _mm_shufflehi_epi16(v_out, _MM_SHUFFLE(3, 1, 2, 0));      // pshufhw
    printf("v_in  = %vhd\n", v_in);
    printf("v_out = %vhd\n", v_out);
    return 0;
}

编译运行:

$ gcc -Wall -msse2 green_goblin_sse2.c 
$ ./a.out 
v_in  = 1 2 3 4 5 6 7 8
v_out = 1 5 2 6 3 7 4 8