c++ - 为什么 std::inner_product 比原始实现慢？

Question

这是我天真的点积实现:

float simple_dot(int N, float *A, float *B) {
    float dot = 0;
    for(int i = 0; i < N; ++i) {
    dot += A[i] * B[i];
    }

    return dot;
}

这是使用 C++ 库:

float library_dot(int N, float *A, float *B) {
    return std::inner_product(A, A+N, B, 0);
}

我运行了一些基准测试(代码在这里 https://github.com/ijklr/sse )，库版本慢了很多。我的编译器标志是 -Ofast -march=native

Answer 1

你的两个函数不做同样的事情。该算法使用一个累加器，其类型是从初始值推导，在您的情况下 (0) 是 int。将 float 累加到 int 不仅比累加到 float 花费的时间更长，而且会产生不同的结果。

原始循环代码的等价物是使用初始值 0.0f，或等价于 float{}。

(请注意，std::accumulate 在这方面非常相似。)