AVX, 단일 정밀도 복소수의 가로 합계?

Question

저는 4 개의 단 정밀도 복소수를 포함하는 256 비트 AVX 레지스터를 실제, 가상, 실제, 가상 등으로 저장했습니다. 현재 전체 256 비트 레지스터를 메모리에 다시 쓰고 거기에서 합산하고 있지만 비효율적 인 것처럼 보입니다.

어떻게 AVX (또는 AVX2) 내장 함수를 사용하여 복소수 수평 합계를 수행 할 수 있습니까? 내장 함수를 사용하여 비교 가능한 효율성을 가진 대답이 없으면 어셈블리를 사용하여 대답을 수락합니다.

편집 : 레지스터에 AR, AI, BR, BI, CR, CI, DR, DI가 포함되어있는 경우 명확하게하려면 복소수 (AR + BR + CR + DR, AI + BI + CI) + DI). 결과가 256 비트 레지스터에 있으면 2 개의 단 정밀도 부동 소수점 수를 추출 할 수 있습니다.

Edit2가이 : 잠재적 인 솔루션,하지만 반드시 최적이 아닌 ... 단지 AVX (안 AVX2)가 필요합니다

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

한 매우 간단한 해결책은 :

__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 }.