使用 SSE 的任意位置 2 输入混洗

Question

我有两个 4 分量向量，我将它们加载到两个 __m128 变量中。 然后我需要洗牌，使结果看起来像这样：

给定：

__m128 mmMin = _mm_load_ps(&glm::vec4(-1.0f,-2.0f,-3.0f,-4.0f)[0]);
__m128 mmMax = _mm_load_ps(&glm::vec4(1.0f,2.0f,3.0f,4.0f)[0]);

我希望随机播放的结果如下所示：

 //    {mmMin.x,mmMax.x,mmMin.x,mmMax.x}

但我认为_mm_shuffle_ps 是不可能的。

来自SSE docs 我看到_mm_shuffle_ps 总是戴着面具 首先从 __m128 的低 2 个分量插入结果 2 个值，然后从高 2 个分量插入 2 个值。

SPU 内部函数具有si_shufb 方法，该方法允许定义基于qword 的掩码并随机播放我希望的任何位置。 SSE有类似的方法吗？

我正在使用 SSE2，但也很高兴看到它可以如何与其他版本（包括 AVX）一起使用。

Answer 1

仅使用 SSE2，我认为您需要 2 次随机播放：unpcklps 交错，然后 unpcklpd same,same 或 shufps same,same 广播低 64 位。

使用 AVX512F，vpermt2ps 可以一次性完成此操作（使用控制向量）；我认为在 AVX2 或更早版本中没有任何 2-source shuffle，在此之前具有足够精细的粒度和灵活的源位置。并且没有固定的随机播放重复元素以及交错。

2-source shuffle 在 AVX512 之前很少见：主要是固定的 shuffle，例如 unpckl/h* 和 palignr。在那之前，它主要是[v]shufps / [v]shufpd。可变控制洗牌也很少见：在 AVX 之前，唯一的一个是 pshufb。 AVX1/2 添加了一些可变控制双字元素随机播放，但仅适用于 1 个源。在 AVX512 之前，没有可变控制 2 源 shuffle。

立即洗牌需要 4 组以上的 2 位索引来处理对两个 4 元素向量串联的任意索引。但是 x86 SIMD 指令始终最多有一个 8 位立即数操作数。 不幸的是，没有像 ARM 这样的广播立即数可以有效地创建 1.0f 或其他值的向量。

---

AVX

由于您只需要每个向量中的 1 个元素，因此您可以使用 AVX 广播加载而不是加载整个向量，然后使用 vblendps

广播负载的成本与英特尔 CPU 上的正常负载相同（不要为 shuffle 端口花费 uop，纯粹在加载端口中处理）。在 AVX512F 之前，它们不能折叠到 ALU 指令的内存操作数中，但它们确实避免了 shuffle-port 瓶颈。 AMD CPU 可能仍需要 ALU uop，但它们有更多的 shuffle ALU，因此 shuffle 吞吐量几乎不是瓶颈。 (https://agner.org/optimize/)

不幸的是，Ryzen vbroadcastss xmm, [mem] 是 2 个独立的前端微指令，但它仍然具有每时钟 2 个吞吐量。

blend-immediate 在 dword 和更高版本的元素上非常有效，可以在 Haswell 及更高版本的任何端口上运行，或者在 SnB/IvB 和 Ryzen 上的 2 个端口上运行。但即使在 Nehalem 上仍然存在单 uop / 1c 延迟。

#include <immintrin.h>
__m128 broadcast_interleave_scalars_avx(const float *min, const float *max) {
    __m128 minx = _mm_broadcast_ss(min);
    __m128 maxx = _mm_broadcast_ss(max);
    return _mm_blend_ps(minx, maxx, 0b1010);
}

On Godbolt，clang 的 asm cmets 确认我得到了正确的混合常量：

    vbroadcastss    xmm0, dword ptr [rdi]
    vbroadcastss    xmm1, dword ptr [rsi]
    vblendps        xmm0, xmm0, xmm1, 10 # xmm0 = xmm0[0],xmm1[1],xmm0[2],xmm1[3]

如果您的数据已经在寄存器中，而不是新加载的，您可能只想使用 2 次随机播放。

---

使用 SSE4.1，您可以执行 2 次 movddup 加载以从内存中广播 64 位（包括您关心的 32 位）然后 blendps。第一次加载将加载超过您关心的float 的 32 位，第二次将加载您关心的 float之前的 32 位。

要让 C++ 编译器为您生成此代码，您必须将指针强制转换为 double* 以加载 __m128d _mm_loaddup_pd (double const* mem_addr)，然后使用 _mm_castpd_ps 从 __m128d 获取 __m128。

https://www.felixcloutier.com/x86/movsldup 也可用于设置 unpcklps。