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

Corrected nvbench.State.getBlockingKernel -> getBlockingKernelTimeout

Browse Source 
Similar change for setBlockingKernelTimeout.

Corrected statement in a comment.
This commit is contained in:
Oleksandr Pavlyk
2025-07-01 12:00:06 -05:00
parent c184549cda
commit c49d718f65
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/src/py_nvbench.cpp
View File

@@ -333,7 +333,7 @@ PYBIND11_MODULE(_nvbench, m)
// nvbench::state::get_short_description
// nvbench::state::exec
// NOTE:
// PyState wraps std::reference_wrapper<nvbench::state>
// State wraps std::reference_wrapper<nvbench::state>
using state_ref_t = std::reference_wrapper<nvbench::state>;
auto pystate_cls = py::class_<nvbench::state>(m, "State");
@@ -400,8 +400,8 @@ PYBIND11_MODULE(_nvbench, m)
pystate_cls.def("getTimeout", &nvbench::state::get_timeout);
pystate_cls.def("setTimeout", &nvbench::state::set_timeout);
pystate_cls.def("getBlockingKernel", &nvbench::state::get_blocking_kernel_timeout);
pystate_cls.def("setBlockingKernel", &nvbench::state::set_blocking_kernel_timeout);
pystate_cls.def("getBlockingKernelTimeout", &nvbench::state::get_blocking_kernel_timeout);
pystate_cls.def("setBlockingKernelTimeout", &nvbench::state::set_blocking_kernel_timeout);
pystate_cls.def("collectCUPTIMetrics", &nvbench::state::collect_cupti_metrics);
pystate_cls.def("isCUPTIRequired", &nvbench::state::is_cupti_required);