Skip to content

Commit c07a84f

Browse files
Corrected nvbench.State.getBlockingKernel -> getBlockingKernelTimeout
Similar change for setBlockingKernelTimeout. Corrected statement in a comment.
1 parent 51fd16e commit c07a84f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

python/src/py_nvbench.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -333,7 +333,7 @@ PYBIND11_MODULE(_nvbench, m)
333333
// nvbench::state::get_short_description
334334
// nvbench::state::exec
335335
// NOTE:
336-
// PyState wraps std::reference_wrapper<nvbench::state>
336+
// State wraps std::reference_wrapper<nvbench::state>
337337

338338
using state_ref_t = std::reference_wrapper<nvbench::state>;
339339
auto pystate_cls = py::class_<nvbench::state>(m, "State");
@@ -400,8 +400,8 @@ PYBIND11_MODULE(_nvbench, m)
400400
pystate_cls.def("getTimeout", &nvbench::state::get_timeout);
401401
pystate_cls.def("setTimeout", &nvbench::state::set_timeout);
402402

403-
pystate_cls.def("getBlockingKernel", &nvbench::state::get_blocking_kernel_timeout);
404-
pystate_cls.def("setBlockingKernel", &nvbench::state::set_blocking_kernel_timeout);
403+
pystate_cls.def("getBlockingKernelTimeout", &nvbench::state::get_blocking_kernel_timeout);
404+
pystate_cls.def("setBlockingKernelTimeout", &nvbench::state::set_blocking_kernel_timeout);
405405

406406
pystate_cls.def("collectCUPTIMetrics", &nvbench::state::collect_cupti_metrics);
407407
pystate_cls.def("isCUPTIRequired", &nvbench::state::is_cupti_required);

0 commit comments

Comments
 (0)