我想在加载大文件后使用Google基准测试库运行多个基准测试。为此,我使用以下代码。函数read_collection()
加载文件的内容,基准Build
处理来自coll
的内容。
#define COLLECTION 'w'
class BuildFixture : public ::benchmark::Fixture {
public:
std::unique_ptr<Collection> coll;
BuildFixture() {
cout << "Constructor\n";
coll = std::make_unique<Collection>(Collection(COLLECTION));
coll->read_collection();
}
~BuildFixture() {
cout << "Destroy collection\n";
coll.reset();
}
};
BENCHMARK_DEFINE_F(BuildFixture, Build1)(benchmark::State& state) {
nrows = static_cast<size_t>(state.range(0));
for (auto _ : state) {
// Do something with coll and nrows
}
}
BENCHMARK_DEFINE_F(BuildFixture, Build2)(benchmark::State& state) {
nrows = static_cast<size_t>(state.range(0));
for (auto _ : state) {
// Something else with coll and nrows
}
}
BENCHMARK_REGISTER_F(BuildFixture, Build1)->Arg(10);
BENCHMARK_REGISTER_F(BuildFixture, Build2)->Arg(20);
BENCHMARK_MAIN();
当我运行此代码时,每个带有参数10和20的基准测试将执行构造函数(总共两次),运行基准测试,然后调用析构函数。所以输出看起来像
Constructor
Constructor
.. (benchmarking outputs)..
Destroy collection
Destroy collection
这最终会花费太多时间来多次读取(相同)文件,并且还会占用额外的内存来为多个基准保存相同的数据。我也担心结果是否会受到页面错误的影响。因此,我有两个问题:
main
功能)更新1
我需要注册的基准是不同的。我不希望将不同的参数传递给同一基准。我已经用Build1
和Build2
相应地更新了问题。
如果定义自己的main
,则可以创建一个全局集合并从每个基准中引用它。
但是,您也可以通过仅注册一次基准测试来避免此问题:
BENCHMARK_REGISTER_F(BuildFixture, Build)->Arg(10)->Arg(20)