使用 SIMD 指令去交错音频通道

Question

我正在实现一个混音器，它在没有 SIMD 指令的情况下运行良好，但很难弄清楚如何将我的声音数据提取到单独的通道中。

我的数据采用交错格式：L0R0 L1R1 L2R2 L3R3... 我以相同的格式将它们加载到 __m128i 中，因此我在寄存器中有 4 个样本。

我希望它们位于不同的通道中：L0L1L2L3 R0R1R2R3。这是我缺少的部分。

所以输入是：8 x i16（4xi32 交错） 我希望输出为 left = 4 x f32 和 right = 4 x f32，然后进行混合。

混合后，我可以交错通道，我得到 L0R0 L1R1 L2R2...：

__m128 *src0 = mixed_channel0;
__m128 *src1 = mixed_channel1;
__m128 *dest = (__m128i *)buffer;

for (u32 sample_index = 0; sample_index < sample_chunk_count; ++sample_index)
{
    __m128 s0 = _mm_load_ps((f32 *)src0++);
    __m128 s1 = _mm_load_ps((f32 *)src1++);

    __m128i l = _mm_cvtps_epi32(s0);
    __m128i r = _mm_cvtps_epi32(s1);

    __m128i lr0 = _mm_unpacklo_epi32(l, r);
    __m128i lr1 = _mm_unpackhi_epi32(l, r);

    *dest++ = _mm_packs_epi32(lr0, lr1);
}

基本上我需要做相反的事情：

__m128i input = [L0R0, L1R1, L2R2, L3R3] packed pairs of 16bit ints
// magic happens, then
__m128 left = [L0, L1, L2, L3] packed 32bit floats
__m128 right = [R0, R1, R2, R3] packed 32bit floats

即使我屏蔽了低/高阶 i16-s，那么如何将它们转换为 f32-s？屏蔽后我会得到：

__m128i right = [xx, R0, xx, R1, xx, R2, xx, R3]
__m128i left = [L0, xx, L1, xx, L2, xx, L3, xx]

如果我可以将它们转换为 4 x i32-s，那么使用 _mm_cvtepi32_ps 将它们转换为 f32-s 会很容易，我就完成了。

谢谢。

Answer 1

从 16 位样本对到 32 位样本的掩码和移位。

// clunky calling convention, but should inline ok.
__m128 unpack_leftright_16bit_channels(__m128i input, __m128 &right_retval) {
    // input = [L0R0, L1R1, L2R2, L3R3] packed pairs of 16bit ints
    __m128i sign_extended_left  = _mm_srai_epi32(input, 16);
    __m128i high_right = _mm_slli_epi32(input, 16);
    __m128i sign_extended_right = _mm_srai_epi32(high_right, 16);

    right_retval = _mm_cvtepi32_ps(sign_extended_right);
    //__m128 right = [R0, R1, R2, R3] packed 32bit floats

    __m128 left  = _mm_cvtepi32_ps(sign_extended_left);
    //__m128 left = [L0, L1, L2, L3] packed 32bit floats
    return left;
}

这个compiles to what you'd expect with gcc5.3，或者clang3.7。

这将成为大多数微架构上 shuffle 吞吐量的瓶颈（请参阅 Agner Fog's insn tables and microarch pdf 和 x86 标签 wiki 中的其他链接）。可能值得使用 SSSE3 pshufb 进行逻辑左移，仅使用实际移位指令进行算术右移，需要在每个 32 位元素的上半部分保留符号位的副本。如果没有 AVX，pshufb 就地洗牌，就像 pslld 就地移动一样（感谢 Intel :()，因此它不会避免额外的 MOV 指令来制作输入的第二个副本。

在 Skylake 上，立即向量移位在 p0/p1 上运行，cvtdq2ps 也是如此。使用pshufb 进行左移可以将吞吐量提高到每个时钟一个浮点输出向量，因为随机播放在端口 5 上运行。

在 Skylake 之前，立即向量移位仅在单个端口上运行，例如哈斯韦尔的 p0。至少这与 int->float 的端口不同：Haswell 在 p1 上运行 cvtdq2ps。同样，pshufb 会将吞吐量提高到每个时钟一个 ps 向量。

---

似乎应该有更好的方法来做到这一点，比如使用 AND 掩码或其他东西。但似乎 2 次移位或 shuffle+shift 是将每个 32 位元素的低 16 位符号扩展为完整 32 位元素的最佳方式。