SSE2 内在函数 - 找到两个无符号短向量的最大值

Question

我想找到包含 8 x 16 位无符号整数元素的两个向量中的最大值。

__m128i vi_A= _mm_loadu_si128(reinterpret_cast<const __m128i*>(&pSrc[0])); // 8 16-Bit Elements
__m128i vi_B= _mm_loadu_si128(reinterpret_cast<const __m128i*>(&pSrc1[0])); // 8 16-Bit Elements
__m128i vi_Max = _mm_max_epi16(vi_A,vi_B);  //<-- Error

但是这个 _mm_max_epi16 是一个带符号的 int 比较，这会导致溢出。所以我尝试通过使用 SSE4.1 intrinsic

来使用它的未签名版本

vi_Max = _mm_max_epu16(vi_A,vi_B);

但不允许我使用 SSE4.1 内在函数。那么找到这些元素最大值的有效方法是什么？

Answer 1

一种（有点低效）的方法是将输入值偏移 0x8000，然后将此偏移量加回结果，例如：

#ifndef __SSE4_1__
inline __m128i _mm_max_epu16(const __m128i v0, const __m128i v1)
{
    return _mm_add_epi16(
               _mm_max_epi16(
                   _mm_sub_epi16(v0, _mm_set1_epi16(0x8000)),
                   _mm_sub_epi16(v1, _mm_set1_epi16(0x8000))),
               _mm_set1_epi16(0x8000));
}
#endif

对于 gcc 或 clang，这会生成一个常量加载指令和四个算术指令。

请注意，您可以使用 _mm_xor_si128 代替 _mm_add_epi16/_mm_sub_epi16，这在意图上可以说更清晰，并且可能会在您的目标架构上提供更好的性能：

#ifndef __SSE4_1__
inline __m128i _mm_max_epu16(const __m128i v0, const __m128i v1)
{
    return _mm_xor_si128(
               _mm_max_epi16(
                   _mm_xor_si128(v0, _mm_set1_epi16(0x8000)),
                   _mm_xor_si128(v1, _mm_set1_epi16(0x8000))),
               _mm_set1_epi16(0x8000));
}
#endif

SSE2 内在函数 - 找到两个无符号短向量的最大值

SSE2 intrinsics - find max of two unsigned short vectors

c++

x86

sse

simd