c++ - x64 上的快速反平方根

Question

我在http://en.wikipedia.org/wiki/Fast_inverse_square_root上找到了网络快速逆平方根。它在 x64 上是否正常工作？有没有人使用并认真测试过？

Answer 1

最初 Fast Inverse Square Root 是为 32 位浮点数编写的，因此只要您对 IEEE-754 浮点表示进行操作，x64 架构就不会影响结果。

请注意，对于“双”精度浮点（64 位），您应该使用另一个常量：

... 64 位 IEEE754 大小类型 double 的“幻数”... 显示为 0x5fe6eb50c7b537a9

Answer 2

这是双精度浮点数的实现：

#include <cstdint>

double invsqrtQuake( double number )
  {
      double y = number;
      double x2 = y * 0.5;
      std::int64_t i = *(std::int64_t *) &y;
      // The magic number is for doubles is from https://cs.uwaterloo.ca/~m32rober/rsqrt.pdf
      i = 0x5fe6eb50c7b537a9 - (i >> 1);
      y = *(double *) &i;
      y = y * (1.5 - (x2 * y * y));   // 1st iteration
      //      y  = y * ( 1.5 - ( x2 * y * y ) );   // 2nd iteration, this can be removed
      return y;
  }

我做了一些测试，它似乎工作正常

Answer 3

是的，如果使用正确的幻数和相应的整数类型，它就可以工作。除了上面的答案之外，这里还有一个适用于double和的 C++11 实现float。条件应该在编译时优化。

template <typename T, char iterations = 2> inline T inv_sqrt(T x) {
    static_assert(std::is_floating_point<T>::value, "T must be floating point");
    static_assert(iterations == 1 or iterations == 2, "itarations must equal 1 or 2");
    typedef typename std::conditional<sizeof(T) == 8, std::int64_t, std::int32_t>::type Tint;
    T y = x;
    T x2 = y * 0.5;
    Tint i = *(Tint *)&y;
    i = (sizeof(T) == 8 ? 0x5fe6eb50c7b537a9 : 0x5f3759df) - (i >> 1);
    y = *(T *)&i;
    y = y * (1.5 - (x2 * y * y));
    if (iterations == 2)
        y = y * (1.5 - (x2 * y * y));
    return y;
}

至于测试，我在我的项目中使用以下doctest ：

#ifdef DOCTEST_LIBRARY_INCLUDED
    TEST_CASE_TEMPLATE("inv_sqrt", T, double, float) {
        std::vector<T> vals = {0.23, 3.3, 10.2, 100.45, 512.06};
        for (auto x : vals)
            CHECK(inv_sqrt<T>(x) == doctest::Approx(1.0 / std::sqrt(x)));
    }
#endif