c - 加载指令中的 AVX2 __m256i const* mem_addr 与 AVX

Question

我无法像之前在 AVX 中那样加载或存储 AVX2 内在函数指令。没有错误，只是警告，并且它不会在运行时执行加载/存储指令。其他 AVX2 指令工作正常，但我无法从内存加载。

如下。

AVX:

float t[MAX][MAX];
row0 = _mm256_load_ps(&t[i][j]);
_mm256_store_ps(&t[j][i], row0);

AVX2:

const int32_t a[MAX][MAX]; // I tried int, long, global and local and many other things... 
a0_i =_mm256_stream_load_si256 (&a[0][0]);
mm256_store_si256(&a[0][0], a0_i);

那么，问题/差异是什么？有什么想法或解决方案吗？

Answer 1

如果您查看 _mm256_stream_load_si256 的原型(prototype):

__m256i _mm256_stream_load_si256 (__m256i const* mem_addr);

您可以看到您需要将其转换为正确的类型，即:

a0_i =_mm256_stream_load_si256 ((__m256i *)&a[0][0]);
                                 ^^^^^^^^^ ^

您还忘记了数组第一个元素的地址，并且在后续存储中还出现了一些进一步的错误:

_mm256_store_si256((__m256i *)&a[0][0], a0_i);
^                   ^^^^^^^^^ 

请注意，当您编译成功后，下一个问题可能是运行时的内存对齐。