2

原始标量函数

static inline uint32 abc(uint32 bytes, int shift)
{
uint32 kMul= 0x1e35a7bd;
return (bytes * kMul) >> shift;

}

等效的 AVX 功能

 static inline uint32 abc(uint32 bytes, int shift)
    {
    uint32 kMul= 0x1e35a7bd;
     __m256i a,b,c,resShift,shift_256i;
    a=_mm256_set1_epi32(bytes);
    b=_mm256_set1_epi32(kMul);
    shift_256i=_mm256_set1_epi32(shift);
    c=_mm256_mul_epi32(a,b);
    resShift=_mm256_srlv_epi64(c,shift_256i);
   // I am not sure what function to use to convert m256i variable into integer
 }

我不确定如何在最后一步将 m256i 变量转换为整数。resShift 具有右移的 m256i 值，但我必须将其转换为整数形式并从该函数返回。有什么帮助吗？

4

1 回答 1

2

您可以查阅英特尔内部指南。您显然需要一些内在的提取物，但我无法一步完成。以下是如何在两个指令中执行此操作：

__m128i lower = _mm256_extracti128_si256(resShift, 0);
return _mm_extract_epi32(lower, 0);

但是请注意，您所做的事情完全没有意义。您不必为单个输入计算单个结果，而是将该输入复制八次并解决八个相同的问题以获得八个相同的答案。所以你不需要做八倍的工作。此外，复制本身（set1_intrinsics）将需要额外的时间。您的 AVX 例程不会比标量例程运行得更快。

PS 顺便说一句，您应该使用_mm256_srlv_epi32而不是_mm256_srlv_epi64进行移位，因为您已经在寄存器中打包了 32 位数据。

于 2015-07-21T04:45:32.427 回答