如何将 __m256i 数据类型存储为整数?
我知道对于花车有:
_mm256_store_ps(float *a, __m256 b)
第一个参数是输出数组。
对于我只找到的整数:
_mm256_store_si256(__m256i *a, __m256i b)
其中两个参数都是 __m256i 数据类型。
做这样的事情是否足够:
int * X = (int*) _mm_malloc( N * sizeof (*X) ,32 );
(我将其用作函数的参数并且我想获取它的值)
内部函数:
__m256i * Xmmtype = (__m256i*) X;
//fill output
_mm256_store_si256( &Xmmtype[ i ] , T ); //T is __m256i
这样可以吗?
-----已更新 ----------------------
好吧,那如果我有呢:
__m256i T;
for ( y = 0; y < h; y++ )
{
for ( x = 0; x < w; x++ )
{
for ( int i = 0; i < N; i+=8 )
{
//calculate here the T
}
//write result
_mm256_store_si256( &Xmmtype[ x + y * w ] , T );
}
}
你所做的没问题,但你不需要创建一个临时指针 - 你可以直接应用强制转换,例如:
_mm256_store_si256( (__m256i *)X, T );
或:
_mm256_store_si256( (__m256i *)&X[i], T );
根据您问题的最新编辑进行更新:
看起来您正在以不符合 AVX 对齐要求的方式对 X
进行索引,即 X[i]
不保证为 32 字节对齐,因此你应该使用未对齐的商店:
_mm256_storeu_si256( (__m256i *)&X[i], T );
我是一名优秀的程序员,十分优秀!