Skip to content

Commit c2a2acc

Browse files
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
1 parent 584f48a commit c2a2acc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

python/src/py_nvbench.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -376,15 +376,15 @@ PYBIND11_MODULE(_nvbench, m)
376376
py::arg("duration_seconds"));
377377
py_benchmark_cls.def(
378378
"set_throttle_threshold",
379-
[](nvbench::benchmark_base &self, nvbench::float64_t threshold) {
379+
[](nvbench::benchmark_base &self, nvbench::float32_t threshold) {
380380
self.set_throttle_threshold(threshold);
381381
return std::ref(self);
382382
},
383383
py::return_value_policy::reference,
384384
py::arg("threshold"));
385385
py_benchmark_cls.def(
386386
"set_throttle_recovery_delay",
387-
[](nvbench::benchmark_base &self, nvbench::float64_t delay) {
387+
[](nvbench::benchmark_base &self, nvbench::float32_t delay) {
388388
self.set_throttle_recovery_delay(delay);
389389
return std::ref(self);
390390
},

0 commit comments

Comments
 (0)