提问者:小点点

如何使用google benchmark对自定义界面进行基准测试


因为除了github自述文件之外,我找不到真正好的资源或文档,所以我会在这里询问社区。

假设我有一个名为的接口和多个派生类型。br>我希望使用相同的函数和传递给的相同参数对所有派生类型的虚函数进行基准测试。br>对于每个派生类型,我希望使用从传递的不同参数在单个基准测试中多次调用

另外,我的一些派生类型的构造函数需要额外的参数,我应该如何处理它呢?

在我看来,最简单的做法是使用模板化基准测试并传递到:

const std::vector<std::size_t> v {1, 2, 4, 8, 16};
BENCHMARK_TEMPLATE(func, Derived1, v);
BENCHMARK_TEMPLATE(func, Derived2, v);
BENCHMARK_TEMPLATE(func, Derived3, v);
...

我觉得这不是正确的方法,因为对于每一个具有不同构造函数签名的派生类型,我需要有不同的基准函数,这几乎与前一个相同。

对此进行基准测试的正确方法是什么?

**更新*:我解决这个问题的方法如下:

std::unique_ptr<Base> GetObj(const std::size_t index)
{
    switch(index)
    {
    case 0:
        return std::make_unique<Derived1>(...);
    case 1:
        return std::make_unique<Derived2>(...);
    ...
    }
}

void func(benchmark::State& state)
{
    auto r = state.range();
    auto object = GetObj(r);
    for (auto _ : state)
    {
        ...
    }
}
BENCHMARK(func)->DenseRange(0,5);

但是我不喜欢的是基准测试的名称是模式,而我希望它是等等…有什么办法实现这个目标吗?


共1个答案

匿名用户

你的方法相当正确。如果您想避免使用多个基准函数,请将基准的核心移到一个单独的函数中,然后从每个per-type函数中调用它。类似于:

static void Benchmark(Base* b, benchmark::State& st) {
  for (auto _ : state) {
    b->DoTheThing()
  }
}

void BM_Foo(benchmark::State& st) {
  Foo f;
  Benchmark(&f, st);
}

BENCHMARK(BM_Foo);

如果您想要更加复杂,可以使用函数更灵活地注册内容。类似于:

auto BM_base = [](benchmark::State& st, auto inputs, Base* b) { /* ... */ };

int main(int argc, char**argv) {
  const std::vector<std::size_t> v = {1, 2, 4, 8, 16};
  benchmark::RegisterBenchmark("foo", BM_base, v, new Foo());
  benchmark::Initialize(&argc, argv);
  return benchmark::RunSpecifiedBenchmarks();
}

顺便说一句,在注册基准时,v是否应该被设置为