c++ - 如何将 __m128 转换为整数

Question

float a[4] = {1,2,3,4}, b[4] = {4,3,2,1};
uint32_t c[4];

int main() {

    __m128 pa = _mm_loadu_ps(a);

    __m128 pb = _mm_loadu_ps(b);
    __m128 pc = _mm_cmpgt_ps(pa, pb);
    _mm_storeu_ps((float*)c, pc);
    for (int i = 0;i  < 4; ++i) printf("%u\n", c[i]);
    return 0;
}

_mm_storeu_ps((float*)c, pc) 的正确指令是什么？ 这里，c是一个整型数组……我觉得这样不好，有什么更好的吗？

Answer 1

在SSE2中有两条指令将__m128(float vector )转换为__m128i(int32_t vector ) : _mm_cvtps_epi32 (四舍五入)和_mm_cvttps_epi32 (有截断)。

__m128i vi = _mm_cvttps_epi32(pc);
_mm_storeu_si128((__m128i *)c, vi);

如果不能使用SSE2，将pc存入floatfloat数组转为int数组数组。

float d[4];
_mm_storeu_ps(d, pc);
c[0] = (int)d[0]; c[1] = (int)d[1]; c[2] = (int)d[2]; c[3] = (int)d[3];