c++ - 如何将参数传递给Google Benchmark计划

Question

我有一个C++ Google Benchmark Program。它使用Google的BENCHMARK_MAIN()方法。现在，我使用Go脚本调用并执行编译后的程序。有没有办法将参数传递到基准程序中？ (我知道主要方法的通用方法，但是我不确定如何在Googletest中进行操作，因为它是在benchmark_api.h中实现的，而我不能只是更改它。)

更新:

到目前为止，我已将宏主体复制到benchmark.cpp中并添加了一行。这不是一个很好的解决方案，因为Google可能对此宏进行了更改(例如更改名称或添加了代码行)不会影响我的副本。它终于在工作了。

int main (int argc, char** argv)
{
    MyNamespace::conf = {argv[1]};
    ::benchmark::Initialize (&argc, argv);
    ::benchmark::RunSpecifiedBenchmarks ();
}

Answer 1

破解整个BENCHMARK_MAIN函数当然是做到这一点的一种方法，但是IMO确实很麻烦而且很丑陋。所以我只想提出一种不同的方法:

// define your chunksize and iteration count combinations here (for i and j) 
static void CustomArguments(benchmark::internal::Benchmark* b) {
  for (int i = 0; i <= 10; ++i)
    for (int j = 0; j <= 50; ++j)
      b->Args({i, j});
}

// the string (name of the used function is passed later)
static void TestBenchmark(benchmark::State& state, std::string func_name) {
  // cout for testing purposes
  std::cout << state.range(0) /* = i */ << " " << state.range(1) /* = j */ 
  << " " << func_name << std::endl;
  for (auto _ : state) {
     // do whatever with i and j and func_name
  }
}

// This macro is used to pass the string "function_name1/2/3" 
// as a parameter to TestBenchmark
BENCHMARK_CAPTURE(TestBenchmark, benchmark_name1, "function_name1")
    ->Apply(CustomArguments);
BENCHMARK_CAPTURE(TestBenchmark, benchmark_name2, "function_name2")
    ->Apply(CustomArguments);
BENCHMARK_CAPTURE(TestBenchmark, benchmark_name3, "function_name3")
    ->Apply(CustomArguments);

BENCHMARK_MAIN()

然后在go脚本中，使用正则表达式过滤器调用基准测试:./prog_name --benchmark_filter=InsertRegexFilterHere
因此，例如:./prog_name --benchmark_filter=TestBenchmark/benchmark_name2/5/35
上面的示例将调用基准测试，并将传递“function_name2”，5和35(这些是您的块大小和迭代计数的值)，因此输出将类似于:

------------------------------------------------------------------------------
Benchmark                                       Time           CPU Iterations
------------------------------------------------------------------------------
TestBenchmark/benchmark_name2/5/35               2 ns          2 ns  308047644