diff --git a/test/benchmark/parse_benchmark_test.cc b/test/benchmark/parse_benchmark_test.cc index 9343cd10..9cf9ad27 100644 --- a/test/benchmark/parse_benchmark_test.cc +++ b/test/benchmark/parse_benchmark_test.cc @@ -54,6 +54,11 @@ void ABSL_ATTRIBUTE_NO_TAIL_CALL BM_ParseCloth(benchmark::State& state) { } BENCHMARK(BM_ParseCloth); +void ABSL_ATTRIBUTE_NO_TAIL_CALL BM_ParseFlag(benchmark::State& state) { + run_parse_benchmark("flag/flag.xml", state); +} +BENCHMARK(BM_ParseFlag); + void ABSL_ATTRIBUTE_NO_TAIL_CALL BM_ParseHumanoid(benchmark::State& state) { run_parse_benchmark("humanoid/humanoid.xml", state); } diff --git a/test/benchmark/step_benchmark_test.cc b/test/benchmark/step_benchmark_test.cc index c4704603..860fabf1 100644 --- a/test/benchmark/step_benchmark_test.cc +++ b/test/benchmark/step_benchmark_test.cc @@ -82,6 +82,13 @@ void ABSL_ATTRIBUTE_NO_TAIL_CALL BM_StepCloth(benchmark::State& state) { } BENCHMARK(BM_StepCloth); +void ABSL_ATTRIBUTE_NO_TAIL_CALL BM_StepFlag(benchmark::State& state) { + MujocoErrorTestGuard guard; + static mjModel* model = LoadModelFromPath("flag/flag.xml"); + run_step_benchmark(model, state); +} +BENCHMARK(BM_StepFlag); + void ABSL_ATTRIBUTE_NO_TAIL_CALL BM_StepHumanoid(benchmark::State& state) { MujocoErrorTestGuard guard; static mjModel* model = LoadModelFromPath("humanoid/humanoid.xml");