File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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);
You can’t perform that action at this time.
0 commit comments