14

我想将浮点值转换为 16 位无符号整数而不饱和（改为环绕/溢出）。

#include <iostream>
#include <xmmintrin.h>

void satur_wrap()
{
    const float bigVal = 99000.f;
    const __m128 bigValVec = _mm_set1_ps(bigVal);

    const __m64 outVec64 =_mm_cvtps_pi16(bigValVec);

#if 0
    const __m128i outVec = _mm_movpi64_epi64(outVec64);
#else

    #if 1
        const __m128i outVec  = _mm_packs_epi32(_mm_cvttps_epi32(bigValVec), _mm_cvttps_epi32(bigValVec));
    #else
        const __m128i outVec  = _mm_cvttps_epi32(bigValVec);
    #endif

#endif

    uint16_t *outVals = NULL;
    posix_memalign((void **) &outVals, sizeof(__m128i), sizeof(__m128i));

    _mm_store_si128(reinterpret_cast<__m128i *>(outVals), outVec);

    for (int i = 0; i < sizeof(outVec) / sizeof(*outVals); i++)
    {
        std::cout << "outVals[" << i << "]: " << outVals[i] << std::endl;
    }

    std::cout << std::endl
        << "\tbigVal: " << bigVal << std::endl
        << "\t(unsigned short) bigVal: " << ((unsigned short) bigVal)  << std::endl
        << "\t((unsigned short)((int) bigVal)): " << ((unsigned short)((int) bigVal)) << std::endl
        << std::endl;
}

示例执行：

$ ./row
outVals[0]: 32767
outVals[1]: 32767
outVals[2]: 32767
outVals[3]: 32767
outVals[4]: 32767
outVals[5]: 32767
outVals[6]: 32767
outVals[7]: 32767

        bigVal: 99000
        (unsigned short) bigVal: 65535
        ((unsigned short)((int) bigVal)): 33464

该((unsigned short)((int) bigVal))表达式按需要工作（但它可能是 UB，对吗？）。但我找不到与 SSE 非常相似的东西。我一定遗漏了一些东西，但我找不到将四个 32-bit 转换为四个 32-bitfloat的原语int。

编辑：糟糕，我认为 32 位整数 -> 16 位无符号整数转换使用环绕是“正常的”。但我从那以后了解到_mm_packs_epi32使用有符号饱和（并且似乎没有 a _mm_packus_epi32）。有没有办法设置模式，或者除此之外的其他原语_mm_packus_epi32？

4

2 回答 2

10

我认为您可能正在寻找CVTTPS2DQ指令，其内在函数是_mm_cvttps_epi32. 请参阅：http: //msdn.microsoft.com/en-us/library/c8c5hx3b (v=vs.71).aspx#vcref_mm_cvttps_epi32

这是一个完整的实现，它采用 2 x SSE 浮点向量并将它们转换为带有环绕的单个打包 8 x 16 位无符号向量：

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

__m128i vec_float_to_short(const __m128 v1, const __m128 v2)
{
    __m128i v1i = _mm_cvttps_epi32(v1);
    __m128i v2i = _mm_cvttps_epi32(v2);
    v1i = _mm_shuffle_epi8(v1i, _mm_setr_epi8(0, 1, 4, 5, 8, 9, 12, 13, 255, 255, 255, 255, 255, 255, 255, 255));
    v2i = _mm_shuffle_epi8(v2i, _mm_setr_epi8(255, 255, 255, 255, 255, 255, 255, 255, 0, 1, 4, 5, 8, 9, 12, 13));
    return _mm_or_si128(v1i, v2i);
}

int main(void)
{
    __m128 v1 = _mm_setr_ps(0.0f, 1.0f, -1.0f, 32767.0f);
    __m128 v2 = _mm_setr_ps(-32768.0f, 32768.0f, 99999.0f, -99999.0f);
    __m128i v3 = vec_float_to_short(v1, v2);

    printf("v1 = %vf\n", v1);
    printf("v2 = %vf\n", v2);
    printf("v3 = %vhu\n", v3);

    return 0;
}

请注意，这使用PSHUFB( _mm_shuffle_epi8) 需要 SSSE3 aka SSE3.5 aka MNI（请参阅tmmintrin.h），因此这仅适用于当前合理的 CPU（过去 6 年左右来自 Intel 的任何 CPU）。

$ gcc -Wall -mssse3 vec_float_to_short.c -o vec_float_to_short
$ ./vec_float_to_short 
v1 = 0.000000 1.000000 -1.000000 32767.000000
v2 = -32768.000000 32768.000000 99999.000000 -99999.000000
v3 = 0 1 65535 32767 32768 32768 34463 31073
$

请注意，并非所有版本的 gcc 都支持vSIMD 向量的 printf 格式说明符（在本例中，我在 OS X 上使用 Apple 的 gcc）。

于 2012-08-25T19:17:42.750 回答
6

我只回答有关 32 位整数 -> 16 位无符号整数转换的部分问题。

由于您需要回绕，因此只需取每个包含 32 位整数的双字的低位字。这些 16 位整数与 16 位未使用数据交错，因此将它们打包成一个连续数组可能会很方便。最简单的方法是使用_mm_shuffle_epi8内在 (SSSE3)。

如果您希望您的程序更具可移植性并且只需要 SSE2 指令集，您可以使用 打包这些值_mm_packs_epi32，但使用以下技巧禁用其饱和行为：

x = _mm_slli_epi32(x, 16);
y = _mm_slli_epi32(y, 16);

x = _mm_srai_epi32(x, 16);
y = _mm_srai_epi32(y, 16);

x = _mm_packs_epi32(x, y);

这个技巧有效，因为它执行 16 位值的符号扩展，这使得有符号饱和成为空操作。

同样的技巧适用于_mm_packus_epi32：

x = _mm_and_si128(x, _mm_set1_epi32(65535));
y = _mm_and_si128(y, _mm_set1_epi32(65535));
x = _mm_packus_epi32(x, y);

这个技巧有效，因为它执行 16 位值的零扩展，这使得无符号饱和成为无操作。执行零扩展更容易，但您需要 SSE4.1 指令集才能_mm_packus_epi32使用。

可以使用一条指令打包 8 个 16 位整数：_mm_perm_epi8. 但这需要非常少见的 XOP 指令集。

这里有几个关于饱和转换的词。

事实上，如果您更改为or _mm_packus_epi32，则内在是可用的。您需要 CPU 和编译器来支持 SSE4.1 扩展。#include <xmmintrin.h>#include <smmintrin.h>#include <x86intrin.h>

如果您没有与 SSE4.1 兼容的 CPU 或编译器，或者希望您的程序更具可移植性，请_mm_packus_epi32使用以下代码替换内部代码：

__m128i m1 = _mm_cmpgt_epi32(x, _mm_set1_epi32(0));
__m128i m2 = _mm_cmpgt_epi32(x, _mm_set1_epi32(65535));
x = _mm_and_si128(x, m1);
x = _mm_or_si128(x, m2);
于 2012-08-28T11:23:08.637 回答