Register proto benchmarks automatically.

andreas-abel · copybara-github · commit 4346cd94f35b · 2023-09-22T10:52:32.000-07:00
PiperOrigin-RevId: 567663491
Change-Id: I4afffd88d84b86f5ba8938d20e60da53a8ab2667
diff --git a/fleetbench/proto/benchmark.cc b/fleetbench/proto/benchmark.cc
@@ -36,7 +36,6 @@ void BM_Protogen_Arena(benchmark::State& state) {
     lifecycle.Run();
   }
 }
-BENCHMARK(BM_Protogen_Arena);
 
 void BM_Protogen_NoArena(benchmark::State& state) {
   const int32_t kIterations = 10;
@@ -48,11 +47,16 @@ void BM_Protogen_NoArena(benchmark::State& state) {
     lifecycle.Run();
   }
 }
-BENCHMARK(BM_Protogen_NoArena);
+
+void RegisterBenchmarks() {
+  REGISTER_BENCHMARK(BM_Protogen_Arena);
+  REGISTER_BENCHMARK(BM_Protogen_NoArena);
+}
 
 class BenchmarkRegisterer {
  public:
   BenchmarkRegisterer() {
+    DynamicRegistrar::Get()->AddCallback(RegisterBenchmarks);
     DynamicRegistrar::Get()->AddDefaultFilter("BM_Protogen_Arena");
   }
 };

Original file line number	Diff line number	Diff line change
`@@ -36,7 +36,6 @@ void BM_Protogen_Arena(benchmark::State& state) {`
`36`	`36`	`lifecycle.Run();`
`37`	`37`	`}`
`38`	`38`	`}`
`39`		`-BENCHMARK(BM_Protogen_Arena);`
`40`	`39`
`41`	`40`	`void BM_Protogen_NoArena(benchmark::State& state) {`
`42`	`41`	`const int32_t kIterations = 10;`
`@@ -48,11 +47,16 @@ void BM_Protogen_NoArena(benchmark::State& state) {`
`48`	`47`	`lifecycle.Run();`
`49`	`48`	`}`
`50`	`49`	`}`
`51`		`-BENCHMARK(BM_Protogen_NoArena);`
	`50`	`+`
	`51`	`+void RegisterBenchmarks() {`
	`52`	`+ REGISTER_BENCHMARK(BM_Protogen_Arena);`
	`53`	`+ REGISTER_BENCHMARK(BM_Protogen_NoArena);`
	`54`	`+}`
`52`	`55`
`53`	`56`	`class BenchmarkRegisterer {`
`54`	`57`	`public:`
`55`	`58`	`BenchmarkRegisterer() {`
	`59`	`+ DynamicRegistrar::Get()->AddCallback(RegisterBenchmarks);`
`56`	`60`	`DynamicRegistrar::Get()->AddDefaultFilter("BM_Protogen_Arena");`
`57`	`61`	`}`
`58`	`62`	`};`