NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-05-13 17:55:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support nvbench.State.set_throttle_threshold

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2025-07-28 14:27:50 -05:00
parent afb9951ed8
commit 413c4a114b
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/src/py_nvbench.cpp
View File

@@ -481,6 +481,9 @@ PYBIND11_MODULE(_nvbench, m)
[](const nvbench::state &state) { return std::ref(state.get_benchmark()); },
py::return_value_policy::reference);
pystate_cls.def("get_throttle_threshold", &nvbench::state::get_throttle_threshold);
pystate_cls.def("set_throttle_threshold",
&nvbench::state::set_throttle_threshold,
py::arg("throttle_fraction"));
pystate_cls.def("get_min_samples", &nvbench::state::get_min_samples);
pystate_cls.def("set_min_samples",