c - SSE 向量化结果错误

Question

下面的代码生成以下输出:

660140021597270387

这意味着只有前两个位置计算正确。但是，我正在处理长整型(4 个字节)，而 __m128i 可以容纳超过 4 个长整型。

long* AA = (long*)malloc(32*sizeof(long));
long* BB = (long*)malloc(32*sizeof(long));

for(i = 0; i<4;i++){
    AA[i] = 2;
    BB[i] = 3;
}

__m128i* m1 = (__m128i*) AA;
__m128i* m2 = (__m128i*) BB;

__m128i m3 = _mm_mul_epu32(m1[0],m2[0]);

long* CC = (long*) malloc(16 * sizeof(long));
CC = (long*)&m3;

for (i = 0; i < 4; i++)
    printf("%ld \n",CC[i]);

分配:

long* AA = (long*) memalign(16 * sizeof(long), 16);

(和其余 vector )生成一个段。过错。有人可以评论一下吗？

谢谢

Answer 1

1) 不要使用不确定大小的类型，例如long，使用特定的固定类型，例如uint32_t

2) 不要使用 malloc - 它不能保证返回 16 字节对齐的内存，请使用 memalign或同等水平*

3) 不要在 C 中转换 malloc(或任何其他函数返回 void *)的结果

4) 无需再分配另一个缓冲区来打印结果

固定代码:

uint32_t* AA = memalign(32*sizeof(uint32_t), 16);
uint32_t* BB = memalign(32*sizeof(uint32_t), 16);

for (i = 0; i < 4; i++){
    AA[i] = 2;
    BB[i] = 3;
}

__m128i* m1 = (__m128i*)AA;
__m128i* m2 = (__m128i*)BB;

__m128i m3 = _mm_mul_epu32(m1[0], m2[0]);    // 2 x 32x32->64 bit unsigned multiplies -> m3

uint64_t* CC = (uint64_t*)&m3;

for (i = 0; i < 2; i++)                      // display 2 x 64 bit result values
    printf("%llu\n", CC[i]);

*请注意，根据您的平台，您可能需要使用 memalign 以外的调用来分配适当对齐的内存，例如posix_memalign、_mm_malloc 或 _aligned_malloc (WIN32)。