256 비트 ps 벡터의 구성 요소에 액세스하는 방법

Question

효율적으로 256 비트 벡터의 요소에 액세스하려면 어떻게해야합니까? 예를 들어

c = _mm256_dp_ps(a, b, 0xff); 

으로 내적을 계산 한 다음 어떻게 값을 액세스 할 수 있습니까? 나는 높은 부분과 낮은 부분 모두를 얻을 필요가 내가 처음과 같은 128 개 비트 부품을 추출 할 필요가 있음을 제대로 이해하지 :

r0 = _mm256_extractf128_ps(c,0); 
r1 = _mm256_extractf128_ps(c,1); 

만 다음 추출 수레 :

_MM_EXTRACT_FLOAT(fr0, r0, 0); 
_MM_EXTRACT_FLOAT(fr1, r1, 0); 

return fr0 + fr1; 

Answer 1

효율적인 방법은 없습니다. dp_ps 작업 자체가 느리고 후속 추출이 느립니다. 일괄 적으로 더 많은 데이터를 처리 할 수 ​​없다면 SSE4 명령어를 사용하여 내적을 계산하고 256 비트보다 128 비트로 작동하는 것이 더 빠릅니다.

Answer 2

글쎄, 당신 단지 메모리에 저장 한 후 스칼라와 함께 일할 수 : 당신은이처럼 256 비트 레지스터의 하단에 상단을 교환하고 추가 할 수

float v[8]; 
*(__m256)(v) = _mm256_dp_ps(a, b, 0xff); 
float result = v[0] + v[4]; 

:

__m256 c = _mm256_dp_ps(a, b, 0xff); 
__m256 d = _mm256_permute2f128_ps(c, c, 1); 
__m256 result = _mm256_add_ps(c, d); 

두 옵션 중 어느 것보다 훨씬 빠를 확률은 4x 8 와이드 도트 제품을 한 번에 줄이고 함께 줄이는 것입니다. 스케치 :

d0 = _mm256_dp_ps(a[0], b[0], 0xff); 
d1 = _mm256_dp_ps(a[1], b[1], 0xff); 
d2 = _mm256_dp_ps(a[2], b[2], 0xff); 
d3 = _mm256_dp_ps(a[3], b[3], 0xff); 

d01 = _mm256_permute_ps(d0, d1, ...); 
d23 = _mm256_permute_ps(d2, d3, ...); 
d= _mm256_permute_ps(d01, d23, ...); 

d0123upper = _mm256_permute2f128_ps(d0123, d0123, 1); 
d = _mm256_add_ps(d0123upper, d0123); // lower 128 bits contain the results of 4 8-wide dot products