NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-03-14 20:27:24 +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

5
python/cuda/nvbench/__init__.pyi
View File

@@ -169,7 +169,10 @@ class State:
"Get Benchmark this configuration is a part of"
...
def get_throttle_threshold(self) -> float:
"Get throttle threshold value"
"Get throttle threshold value, as fraction of maximal frequency"
...
def set_throttle_threshold(self, threshold_fraction: float) -> None:
"Set throttle threshold fraction to specified value, expected to be between 0 and 1"
...
def get_min_samples(self) -> int:
"Get the number of benchmark timings NVBench performs before stopping criterion begins being used"

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",