NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-05-13 17:55:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

add comments to body of launcher_fn lambda in State.exec method

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2025-07-21 10:57:28 -05:00
parent aa2b4d9960
commit 6b4da8c5cb
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
python/src/py_nvbench.cpp
View File

@@ -449,9 +449,13 @@ PYBIND11_MODULE(_nvbench, m)
pystate_cls.def(
"exec",
[](nvbench::state &state, py::object fn, bool batched, bool sync) {
auto launcher_fn = [fn](nvbench::launch &launch_descr) -> void {
fn(py::cast(std::ref(launch_descr), py::return_value_policy::reference));
[](nvbench::state &state, py::object callable_fn, bool batched, bool sync) {
// wrapper to invoke Python callable
auto launcher_fn = [callable_fn](nvbench::launch &launch_descr) -> void {
// cast C++ object to python object
auto launch_pyarg = py::cast(std::ref(launch_descr), py::return_value_policy::reference);
// call Python callable
callable_fn(launch_pyarg);
};
if (sync)