随机抽取16位向量SSE

时间:2016-03-11 09:37:55

标签: sse simd

我在这里工作SSE和新手。我正在尝试使用shuffle指令来重排一个16位向量,如下所示:

输入:

1 2 3 4 5 6 7 8 

输出:

1 5 2 6 3 7 4 8

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

1 个答案:

答案 0 :(得分:5)

只要您可以假设SSSE3,那么您可以使用pshufb又称_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
相关问题