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

时间:2017-01-12 10:24:14

标签: c++ x86 sse simd

我想找到最多两个包含8 x 16位无符号int元素的向量。

__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内在

来使用它的无符号版本
vi_Max = _mm_max_epu16(vi_A,vi_B);

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

1 个答案:

答案 0 :(得分:5)

一种(效率有点低)的方法是将输入值偏移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