代码之家  ›  专栏  ›  技术社区  ›  Green goblin

乱序16位矢量SSE

  •  3
  • Green goblin  · 技术社区  · 9 年前

    我正在努力 SSE 还有一个新手。我正在尝试使用 shuffle 对16位矢量进行混洗的指令如下:

    输入:

    1 2 3 4 5 6 7 8 
    

    输出:

    1 5 2 6 3 7 4 8
    

    我如何实现期望的目标?我对这里使用的常量感到困惑,我看不到任何16位 洗牌 指示 洗牌 指令仅适用于8位和32位。

    1 回复  |  直到 9 年前
        1
  •  5
  •   Paul R    9 年前

    只要您可以假设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