Question

我有一个256位AVX寄存器，包含4个单精度复数，存储为实数，虚数，实数，虚数等。我正在将整个256位寄存器写回存储器并将其求和，但这似乎效率低下。

如何使用AVX（或AVX2）内在函数执行复数水平求和？如果没有使用内在函数的效率相当的答案，我会接受使用汇编的答案。

编辑：为了澄清，如果寄存器包含AR，AI，BR，BI，CR，CI，DR，DI，我想计算复数（AR + BR + CR + DR，AI + BI + CI + DI）。如果结果是256位寄存器，我可以提取2个单精度浮点数。

Edit2：潜在的解决方案，虽然不一定是最佳的......

float hsum_ps_sse3(__m128 v) {
    __m128 shuf = _mm_movehdup_ps(v);        // broadcast elements 3,1 to 2,0
    __m128 sums = _mm_add_ps(v, shuf);
    shuf        = _mm_movehl_ps(shuf, sums); // high half -> low half
    sums        = _mm_add_ss(sums, shuf);
    return        _mm_cvtss_f32(sums);
}

float sumReal = 0.0;
float sumImaginary = 0.0;

 __m256i mask = _mm256_set_epi32 (7, 5, 3, 1, 6, 4, 2, 0);

 // Separate real and imaginary.
__m256 permutedSum = _mm256_permutevar8x32_ps(sseSum0, mask);
__m128 realSum = _mm256_extractf128_ps(permutedSum , 0);
__m128 imaginarySum = _mm256_extractf128_ps(permutedSum , 1);

// Horizontally sum real and imaginary.
sumReal = hsum_ps_sse3(realSum);
sumImaginary = hsum_ps_sse3(imaginarySum);

Answer 1

一个相当直接的解决方案，只需要AVX（不是AVX2）：

__m128i v0 = _mm256_castps256_ps128(v);      // get low 2 complex values
__m128i v1 = _mm256_extractf128_ps(v, 1);    // get high 2 complex values
v0 = _mm_add_ps(v0, v1);                     // add high and low
v1 = _mm_shuffle_ps(v0, v0, _MM_SHUFFLE(1, 0, 3, 2));
v0 = _mm_add_ps(v0, v1);                     // combine two halves of result

结果将在v0中{ sum.re, sum.im, sum.re, sum.im }。

AVX，单精度复数的水平和？

1 个答案: