From c2a2acc9b666a6cf69678461090574fb160a82fd Mon Sep 17 00:00:00 2001 From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com> Date: Mon, 4 Aug 2025 12:14:52 -0500 Subject: [PATCH] Change float64_t arg-type for set_throttle_threshold to float32_t The C++ method signature of set_throttle_threshold/set_trottle_recovery_delay, which uses nvbench::float32_t --- python/src/py_nvbench.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/src/py_nvbench.cpp b/python/src/py_nvbench.cpp index 22cfbc6..8856e8e 100644 --- a/python/src/py_nvbench.cpp +++ b/python/src/py_nvbench.cpp @@ -376,7 +376,7 @@ PYBIND11_MODULE(_nvbench, m) py::arg("duration_seconds")); py_benchmark_cls.def( "set_throttle_threshold", - [](nvbench::benchmark_base &self, nvbench::float64_t threshold) { + [](nvbench::benchmark_base &self, nvbench::float32_t threshold) { self.set_throttle_threshold(threshold); return std::ref(self); }, @@ -384,7 +384,7 @@ PYBIND11_MODULE(_nvbench, m) py::arg("threshold")); py_benchmark_cls.def( "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); return std::ref(self); },