Question

我正在将一些代码从SSE升级到AVX2。一般来说，我可以看到收集指令非常有用并且有益于性能。但是我遇到了一种情况，即收集指令的效率低于将收集操作分解为更简单的操作。

在下面的代码中，我有一个int32 b向量，一个double xi向量和4个int32索引，打包在128位寄存器bidx中。我需要先从向量b收集，而不是从向量xi收集。即，在伪代码中，我需要做：

__m128i i = b[idx];
__m256d x = xi[i];

在下面的函数中，我使用#ifdef以两种方式实现这一点：通过收集指令，产生290 Miter / sec的吞吐量并通过基本操作，产生325 Miter / sec的吞吐量。

有人可以解释发生了什么吗？感谢

inline void resolve( const __m256d& z, const __m128i& bidx, int32_t j
                    , const int32_t *b, const double *xi, int32_t* ri )
{

    __m256d x;
    __m128i i;

#if 0  // this code uses two gather instructions in sequence

    i = _mm_i32gather_epi32(b, bidx, 4));  // i = b[bidx]
    x = _mm256_i32gather_pd(xi, i, 8);     // x = xi[i]

#else  // this code does not use gather instructions

    union {
            __m128i vec;
            int32_t i32[4];
    } u;
    x = _mm256_set_pd
            ( xi[(u.i32[3] = b[_mm_extract_epi32(bidx,3)])]
            , xi[(u.i32[2] = b[_mm_extract_epi32(bidx,2)])]
            , xi[(u.i32[1] = b[_mm_extract_epi32(bidx,1)])]
            , xi[(u.i32[0] = b[_mm_cvtsi128_si32(bidx)  ])]
            );
    i = u.vec;

#endif

    // here we use x and i
    __m256  ps256 = _mm256_castpd_ps(_mm256_cmp_pd(z, x, _CMP_LT_OS));
    __m128  lo128 = _mm256_castps256_ps128(ps256);
    __m128  hi128 = _mm256_extractf128_ps(ps256, 1);
    __m128  blend = _mm_shuffle_ps(lo128, hi128, 0 + (2<<2) + (0<<4) + (2<<6));
    __m128i lt    = _mm_castps_si128(blend);  // this is 0 or -1
    i = _mm_add_epi32(i, lt);
    _mm_storeu_si128(reinterpret_cast<__m128i*>(ri)+j, i);
}

Answer 1

因为您的决定＆＃39;函数被标记为内联我想它是在高频循环中调用的。然后，您可能还会看到“解决方案”之外的输入参数之间的依赖关系。功能。在使用标量代码变体时，编译器可能能够更好地跨循环边界优化内联代码。

为什么两个连续收集指令的表现比同等的基本操作差？

1 个答案: