有没有一种使用SSE找到两个变量模数的好方法? (没有SVML)

时间:2014-07-14 19:50:05

标签: c++ sse

我试图学习使用SSE,我正在制作的程序之一需要使用模数除法,因此我写了这样做(抱歉它被过度评价):

__m128i SSEModDiv(__m128i input, __m128i divisors)
{
    //Error Checking (div by zero)
    /*__m128i zeros = _mm_set1_epi32(0);
    __m128i error = _mm_set1_epi32(-1);
    __m128i zerocheck = _mm_cmpeq_epi32(zeros, divisors);
    if (_mm_extract_epi16(zerocheck, 0) != 0)
        return error;
    if (_mm_extract_epi16(zerocheck, 2) != 0)
        return error;
    if (_mm_extract_epi16(zerocheck, 4) != 0)
        return error;
    if (_mm_extract_epi16(zerocheck, 6) != 0)
        return error;*/

    //Now for the real work
    __m128 inputf = _mm_cvtepi32_ps(input);
    __m128 divisorsf = _mm_cvtepi32_ps(divisors);

    /*__m128 recip = _mm_rcp_ps(divisorsf); //Takes reciprocal
    __m128 divided = _mm_mul_ps(inputf, recip); //multiplies by reciprical values*/
    __m128 divided = _mm_div_ps(inputf, divisorsf);
    __m128i intermediateint = _mm_cvttps_epi32(divided); //makes an integer version truncated
    __m128 intermediate = _mm_cvtepi32_ps(intermediateint);
    __m128 multiplied = _mm_mul_ps(intermediate, divisorsf); //multiplies the intermediate with the divisors
    __m128 mods = _mm_sub_ps(inputf, multiplied); //subtracts to get moduli
    return _mm_cvtps_epi32(mods);
}

问题在于,这与在释放时单独取四个32位整数的每个元素的模数一样快,在调试中找到的速度大约慢10倍(通过分析找到)。

有人能给我任何关于如何加快这项功能的指示吗?

- 我不能使用SVML,因为我使用的是Visual Studio -

1 个答案:

答案 0 :(得分:6)

对于inputdivisors的一般值,没有有用的整数除法或模数的SIMD x86指令,因此最好使用标量整数除法。但是,有一些特殊情况可以更快地完成SIMD整数模数。

例如,如果你想做(a + b)%c并且a和b已经减少(即a<cb<c)那么你可以使用这样的比较和减法:

z = a + b
if(z>=c) z-=c;

我在这个例子vectorizing-modular-arithmetic

中做了这个

另一个例子是如果除数不是编译时常数,但在循环中仍然是常数,那么你可以使用浮点除法的类似想法。浮点除法的一个常见技巧是预先计算除数的倒数并进行乘法运算:

float fact = 1.0/x;
for(int i=0; i<n; i++) {
    z[i] = fact*y[i];  //z[i] = y[i]/x;
}

您可以使用类似的整数除法技术将整数除法转换为带移位的整数乘法。

y / x ≈ y * (2 n / x) >> n

有几种不同的技术可以确定因子(又名magic number(2 n / x)和班次n。事实上,大多数编译器已经为编译时常量和除法做了这个。如果您尝试例如x/7并查看GCC或MSVC的程序集输出,您将看到它们实际上没有进行整数除法,它们使用相同的幻数和移位进行乘法和移位。来自http://www.hackersdelight.org/magic.htm

我在运行时使用Agner Fog的Vector Class Library或他的Subroutine library执行此操作。两个库都会在运行时为您计算SSE和AVX整数除法的幻数和移位。

但正如我在答案开始时所说的,如果你想做的话

for(int i=0; i<n; i++) {
    z[i] = y[i]%x[i];
}

x[i]在循环中不是一个常数,它最好坚持标量除法/模数。