Question

考虑我有一个带有16位数据的SSE数组：

{1,2,3,4,5,6,7,8}

现在我需要通过在前8个字节中仅存储16位数据的低字节来将此SSE数组转换为8位数据，如：

{1,2,3,4,5,6,7,8,0,0,0,0,0,0,0,0}.

是否有任何SSE指令来执行此操作？

Answer 1

正如@harold在上面的评论中所说，您可以使用pshufb aka _mm_shuffle_epi8轻松完成此操作，例如

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

static __m128i pack_16_to_8(const __m128i v)
{
    const __m128i vperm = _mm_setr_epi8(0, 2, 4, 6, 8, 10, 12, 14, -1, -1, -1, -1, -1, -1, -1, -1);

    return _mm_shuffle_epi8(v, vperm);
}

int main(void)
{
    const __m128i v = _mm_setr_epi16(1, 2, 3, 4, 5, 6, 7, 8);

    printf("%vhd -> %vd\n", v, pack_16_to_8(v));
    return 0;
}

编译并运行：

$ gcc -Wall -mssse3 pack_16_to_8.c && ./a.out

1 2 3 4 5 6 7 8 -> 1 2 3 4 5 6 7 8 0 0 0 0 0 0 0 0

Answer 2

除了保罗К的回答：

SSE2扩展包含命令PACKSSWB(_mm_packs_epi16)和PACKUSWB (_mm_packus_epi16)。这些命令专门用于将16位向量转换为8位向量。如果这些值超出范围8位无符号整数（0..255），它们将执行16位（有符号和无符号）值的饱和。

#include <iostream>
#include <emmintrin.h>

template<class T> inline void Print(const __m128i & v)
{
    T b[sizeof(v) / sizeof(T)];
    _mm_storeu_si128((__m128i*)b, v);
    for (int i = 0; i < sizeof(v) / sizeof(T); i++)
        std::cout << int(b[i]) << " ";
    std::cout << std::endl;
}

int main()
{
    __m128i v16 = _mm_setr_epi16(1, 2, 3, 4, 5, 6, 7, 8);

    Print<uint8_t>(_mm_packs_epi16(v16, _mm_setzero_si128()));
    Print<uint8_t>(_mm_packus_epi16(v16, _mm_setzero_si128()));

    return 0;
}

输出：

1 2 3 4 5 6 7 8 0 0 0 0 0 0 0 0
1 2 3 4 5 6 7 8 0 0 0 0 0 0 0 0

将8个16位SSE寄存器转换为8位数据

2 个答案: