NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-04-30 19:51:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change float64_t arg-type for set_throttle_threshold to float32_t

Browse Source 
The C++ method signature of set_throttle_threshold/set_trottle_recovery_delay,
which uses nvbench::float32_t
This commit is contained in:
Oleksandr Pavlyk
2025-08-04 12:14:52 -05:00
parent 584f48ac97
commit c2a2acc9b6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/src/py_nvbench.cpp
View File

@@ -376,7 +376,7 @@ PYBIND11_MODULE(_nvbench, m)
py::arg("duration_seconds")); py::arg("duration_seconds"));
py_benchmark_cls.def( py_benchmark_cls.def(
"set_throttle_threshold", "set_throttle_threshold",
[](nvbench::benchmark_base &self, nvbench::float64_t threshold) { [](nvbench::benchmark_base &self, nvbench::float32_t threshold) {
self.set_throttle_threshold(threshold); self.set_throttle_threshold(threshold);
return std::ref(self); return std::ref(self);
}, },
@@ -384,7 +384,7 @@ PYBIND11_MODULE(_nvbench, m)
py::arg("threshold")); py::arg("threshold"));
py_benchmark_cls.def( py_benchmark_cls.def(
"set_throttle_recovery_delay", "set_throttle_recovery_delay",
[](nvbench::benchmark_base &self, nvbench::float64_t delay) { [](nvbench::benchmark_base &self, nvbench::float32_t delay) {
self.set_throttle_recovery_delay(delay); self.set_throttle_recovery_delay(delay);
return std::ref(self); return std::ref(self);
}, },