Skip to content

Commit af32e3f

Browse files
authored
run ClearRegisteredBenchmarks at exit (google#1463)
1 parent a476d0f commit af32e3f

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

bindings/python/google_benchmark/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@ def my_benchmark(state):
2626
if __name__ == '__main__':
2727
benchmark.main()
2828
"""
29+
import atexit
2930

3031
from absl import app
3132
from google_benchmark import _benchmark
@@ -158,3 +159,4 @@ def main(argv=None):
158159
# Methods for use with custom main function.
159160
initialize = _benchmark.Initialize
160161
run_benchmarks = _benchmark.RunSpecifiedBenchmarks
162+
atexit.register(_benchmark.ClearRegisteredBenchmarks)

bindings/python/google_benchmark/benchmark.cc

+1
Original file line numberDiff line numberDiff line change
@@ -179,5 +179,6 @@ PYBIND11_MODULE(_benchmark, m) {
179179
py::return_value_policy::reference);
180180
m.def("RunSpecifiedBenchmarks",
181181
[]() { benchmark::RunSpecifiedBenchmarks(); });
182+
m.def("ClearRegisteredBenchmarks", benchmark::ClearRegisteredBenchmarks);
182183
};
183184
} // namespace

0 commit comments

Comments
 (0)