c - __m256d 的广播条目

Question

使用 AVX 将 __m256d 寄存器的单个条目广播到其他 __m256d 寄存器的所有元素的最快方法是什么？

对于单精度，这可以通过一次调用 _mm256_shuffle_ps() 来完成。此外，对于 AVX2 _mm256d_permute4x64_pd 似乎确实可以解决问题。谢谢。

Answer 1

好的，我想出了一个紧凑的解决方案，但我不太确定这是否是最快的。

double pos0[4] = {0,1,2,3};
__m256d a= _mm256_loadu_pd(pos0);

__m256d a02,a13;
a02 = _mm256_shuffle_pd(a,a,0);
__m256d a0 = _mm256_blend_pd(a02, _mm256_permute2f128_pd(a02,a02,1), 0x0C);
__m256d a2 = _mm256_blend_pd(a02, _mm256_permute2f128_pd(a02,a02,1), 0x03);
a13 = _mm256_shuffle_pd(a,a,0x0F);
__m256d a1 = _mm256_blend_pd(a13, _mm256_permute2f128_pd(a13,a13,1), 0x0C);
__m256d a3 = _mm256_blend_pd(a13, _mm256_permute2f128_pd(a13,a13,1), 0x03);

这会导致:

a0 = {0,0,0,0}
a1 = {1,1,1,1}
a2 = {2,2,2,2}
a3 = {3,3,3,3}