Comment additionner M256 horizontalement?
j'aimerais faire la somme horizontale des composantes d'un vecteur __m256
en utilisant les instructions AVX.
En SSE je pourrais utiliser
_mm_hadd_ps(xmm,xmm);
_mm_hadd_ps(xmm,xmm);
pour obtenir le résultat au premier composant du vecteur, mais cela ne change pas avec la version 256 bits de la fonction ( _mm256_hadd_ps
).
Quelle est la meilleure façon de calculer la somme horizontale d'un vecteur __m256
?
10
demandé sur
Ben-Uri
2012-11-04 17:55:49
2 réponses
cette version devrait être optimale à la fois pour Intel Sandy / Ivy Bridge et AMD Bulldozer:
// x = ( x7, x6, x5, x4, x3, x2, x1, x0 )
float sum8(__m256 x) {
// hiQuad = ( x7, x6, x5, x4 )
const __m128 hiQuad = _mm256_extractf128_ps(x, 1);
// loQuad = ( x3, x2, x1, x0 )
const __m128 loQuad = _mm256_castps256_ps128(x);
// sumQuad = ( x3 + x7, x2 + x6, x1 + x5, x0 + x4 )
const __m128 sumQuad = _mm_add_ps(loQuad, hiQuad);
// loDual = ( -, -, x1 + x5, x0 + x4 )
const __m128 loDual = sumQuad;
// hiDual = ( -, -, x3 + x7, x2 + x6 )
const __m128 hiDual = _mm_movehl_ps(sumQuad, sumQuad);
// sumDual = ( -, -, x1 + x3 + x5 + x7, x0 + x2 + x4 + x6 )
const __m128 sumDual = _mm_add_ps(loDual, hiDual);
// lo = ( -, -, -, x0 + x2 + x4 + x6 )
const __m128 lo = sumDual;
// hi = ( -, -, -, x1 + x3 + x5 + x7 )
const __m128 hi = _mm_shuffle_ps(sumDual, sumDual, 0x1);
// sum = ( -, -, -, x0 + x1 + x2 + x3 + x4 + x5 + x6 + x7 )
const __m128 sum = _mm_add_ss(lo, hi);
return _mm_cvtss_f32(sum);
}
7
répondu
Marat Dukhan
2012-11-04 20:12:20
cela peut être fait avec le code suivant:
ymm2 = _mm256_permute2f128_ps(ymm , ymm , 1);
ymm = _mm256_add_ps(ymm, ymm2);
ymm = _mm256_hadd_ps(ymm, ymm);
ymm = _mm256_hadd_ps(ymm, ymm);
mais il pourrait y avoir une meilleure solution.
3
répondu
Ben-Uri
2012-11-04 15:58:29