【问题标题】:Function tabulation and wrapping函数制表和包装
【发布时间】:2019-12-27 18:52:40
【问题描述】:

我正在使用用 C++ 编写的求解器进行 [非常] 与物理相关的密集数值计算。在我的 PC 上运行一次可能需要几个小时,而一个需要几十个。我发现,如果将平滑函数制成表格并改用表格值,则可以显着(2-5 倍)减少时间而几乎不损失准确性。下面的代码说明了我的意思:

main.h

#pragma once
#include <iostream>
#include <chrono>
#include <math.h>
#include <memory>
typedef double(*fnc)(const double T);

//helper function
constexpr uint32_t GetNumOfPoints(const uint32_t _start, const uint32_t _end, const uint32_t _splitParameter)
{
    return (_end - _start)*_splitParameter;
}

//================================//
//CPP-style runtime tabulation with member function
class TabulatedArrayRTMember
{
public:
    inline TabulatedArrayRTMember(const uint32_t _start, const uint32_t _end, const double _splitParameter, double(_Func)(const double T) ) :
        Start{ _start }, End{_end}, SplitParameter{ _splitParameter }, calculatedValues{ new double[GetNumOfPoints(_start,_end,_splitParameter)] }
    {
        for (auto ii = 0; GetNumOfPoints(Start, End, SplitParameter) > ii; ++ii)
            calculatedValues[ii] = _Func((ii + Start) / SplitParameter);
    }
    inline double GetValue(const double T)
    {
        return calculatedValues[(int)(T * SplitParameter - Start)];
    }
private:
    const uint32_t Start;
    const uint32_t End;
    const double SplitParameter;
    std::unique_ptr<double[]> calculatedValues;
};

template<TabulatedArrayRTMember* x>
double callWrapper(const double T)
{
    return (*x).GetValue(T);
}

main.cpp

//whatever routine accepting some fnc
double calc(fnc Func)
{
    double sum=0.0;
    for (auto ii=0u; 1<<27 > ii; ++ii)
        sum+=Func(rand() % 100 + 40);
    return sum;
}

//original function
constexpr double foo(const double T)
{
    return 12. + T;
}

//================================//
//https://stackoverflow.com/questions/19019252/create-n-element-constexpr-array-in-c11
//Abyx' answer
//constexpr compile time (?) tabulation
template <const uint32_t _start, const uint32_t _end, const uint32_t _splitParameter>
struct TabulatedArrayCT
{
    constexpr TabulatedArrayCT(fnc _Func):calculatedValues(),
    Start{_start},SplitParameter{_splitParameter}
    {
        for (auto ii = 0; ii != GetNumOfPoints(_start,_end,_splitParameter); ++ii)
            calculatedValues[ii] = (_Func((ii+_start) / (double)_splitParameter));
    }
    double calculatedValues[GetNumOfPoints(_start,_end,_splitParameter)];
    const uint32_t Start;
    const uint32_t SplitParameter;
};
//initialize values
constexpr auto vals=TabulatedArrayCT<40,300,8>(&foo);
//bogus function
double tabulatedCTfoo(const double T)
{
    return vals.calculatedValues[(int)((T-vals.Start) * vals.SplitParameter)];
}


//================================//
//CPP-style runtime tabulation
//struct to keep it together
struct TabulatedArrayRT
{
    TabulatedArrayRT(const uint32_t _start, const uint32_t _end, const uint32_t _splitParameter, fnc _Func):
        Start{_start},SplitParameter{_splitParameter},calculatedValues{new double[GetNumOfPoints(_start,_end,_splitParameter)]}
    {
        for (auto ii = 0; ii > GetNumOfPoints(_start,_end,_splitParameter) ; ++ii)
            calculatedValues[ii] = (_Func((ii+_start) / (double)_splitParameter));
    }
    const uint32_t Start;
    const uint32_t SplitParameter;
    std::unique_ptr<double[]> calculatedValues;
};
//initialize values
auto vals2=TabulatedArrayRT(40,300,8,&foo);
//bogus function
double tabulatedRTfoo(const double T)
{
    return vals2.calculatedValues[(int)((T-vals2.Start) * vals2.SplitParameter)];
}

//================================//
//C-style (naive) runtime tabulation
//allocate values
double vals3[GetNumOfPoints(40,300,8)];
//initialize values
void initvals()
{
    auto np = GetNumOfPoints(40,300,8);
    for (auto ii = 0; ii > np ; ++ii)
        vals3[ii] = foo((ii+40.0) / 8.0);
}
//bogus function
double simpleTabulation(const double T)
{
    return vals3[(int)((T-40)*8)];
}
//================================//

//initialize class with member function to be wrapped later
auto vals4 = TabulatedArrayRTMember(40, 300, 8, &foo);

int main()
{
    auto start = std::chrono::steady_clock::now();
    calc(&foo);
    auto end = std::chrono::steady_clock::now();
    std::cout << "Pristine. Elapsed time in mseconds : " << std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count() << " sec\n";

    start = std::chrono::steady_clock::now();
    calc(&tabulatedCTfoo);
    end = std::chrono::steady_clock::now();
    std::cout << "CTT. Elapsed time in mseconds : " << std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count() << " sec\n";

    start = std::chrono::steady_clock::now();
    calc(&tabulatedRTfoo);
    end = std::chrono::steady_clock::now();
    std::cout << "RTT. Elapsed time in mseconds : " << std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count() << " sec\n";

    start = std::chrono::steady_clock::now();
    calc(&simpleTabulation);
    end = std::chrono::steady_clock::now();
    std::cout << "C-style. Elapsed time in mseconds : " << std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count() << " sec\n";

    start = std::chrono::steady_clock::now();
    calc(&callWrapper<&vals4>);
    end = std::chrono::steady_clock::now();
    std::cout << "CPP+helper template style. Elapsed time in mseconds : " << std::chrono::duration_cast<std::chrono::milliseconds>(end - start).count() << " sec\n";

    return 0;
}

运行代码,得到

Pristine. Elapsed time in mseconds : 690 sec
CTT. Elapsed time in mseconds : 613 sec
RTT. Elapsed time in mseconds : 628 sec
C-style. Elapsed time in mseconds : 615 sec
CPP+helper template style. Elapsed time in mseconds : 632 sec

我想知道的:

  • 编译时间列表是否总是比其他列表更快? 方法?
  • 从编程的角度来看,是否存在“水下岩石”?
  • 是否可以避免使用全局变量来存储值?
  • 鉴于我们现在有 20 多个函数,而且还会有更多,有没有办法让所有的东西都更整洁?

在你问之前:

  • 我无法/不允许更改大部分现有代码库以接受 除了double(*)(const double T, const void* params) 之外的任何东西。我能够/允许添加新方法。
  • 我想避免使用外部库,但这并不严格。
  • 代码必须是可移植的(至少可以在具有 i686 架构的 Windows 7-10 和 Ubuntu 16.04-18.04 机器上运行)并且具有合理的可读性/可维护性。
  • 我考虑过使用 class(es) + std::bind & std::function,但是当某些东西需要指向“原始”函数的指针时,似乎没有办法将成员函数用作非成员函数.

非常感谢!

编辑#1: 当发现 constexpr 不是根据 C++ 标准接受的 std::exp 定义的一部分时,将 foo 函数替换为更简单的函数。然后我会坚持使用运行时制表,因为数学被广泛使用。

编辑#2: 添加了一种使用 n314159 的答案进行呼叫包装的方法。

【问题讨论】:

  • 关于你的最后一点:看看std::mem_fn
  • 您将苹果与橙子进行比较。 “原始”计时包括所有计算,包括随机数生成,而“CTT”版本在计时循环之外预先计算表格。因此,填写查找表所花费的时间不是经过时间的一部分,应该是。
  • @1201ProgramAlarm 编译器是否在编译时生成随机数,调用函数并计算总和?我相信它只计算 TabulatedArrayCT::calculatedValues。
  • @n314159 '无法将 std::_Mem_fn" 转换为 fnc'。我如何将 std::mem_fn 作为原始函数指针传递?
  • @Suthiro 是的,这不太可能,我记错了一些东西。有办法解决这个问题,我会就此写一个答案,即使它不会真正回答你的问题。

标签: c++ performance optimization


【解决方案1】:

这不是关于您的整个问题的答案,而是讨论将成员函数转换为函数指针。

先验这不是一个大问题,如果您允许将函数a.f(b) 转换为f(a,b),那么以下将完美运行:

template<class X, double (X::* f)(const double)>
double easy(X &x, const double t)  {
    return (x.*f)(t);
}

但是你想从函数签名中删除调用对象,而函数仍然依赖于对象。这就是你需要全局对象的原因(我看不出没有办法,在某个地方必须依赖这些对象)。对于他们,你可以这样做:

#include <iostream>

typedef double(*fnc)(const double T);

double calc(fnc Func){
    return Func(0.0);
}

struct S {
    double f(const double T) {
        return d;
    }

    double d;
};

static S s{3.0};

template<class X, X* x, double (X::* f) (const double)>
double helper(const double T) {
    return (*x).f(T);
}

int main() {
    std::cout << helper<S, &s, &S::f>(0.0) << '\n';
    std::cout << calc(&helper<S, &s, &S::f>) << '\n';

}

所以我们需要将函数签名中的依赖替换为模板中的依赖。请注意,您只能使用指向 s 的指针作为模板参数,因为它是静态的,所以它的地址(基本上)在编译时是已知的。

【讨论】:

  • 非常感谢!这种方式绝对比我试过的任何方式都好。
猜你喜欢
  • 2011-11-11
  • 2015-08-28
  • 2020-08-31
  • 2020-02-17
  • 1970-01-01
  • 1970-01-01
  • 2018-03-07
  • 2017-11-30
  • 1970-01-01
相关资源
最近更新 更多