Add nvbench.Benchmark.set_run_once method

2026-05-13 17:55:39 +00:00 · 2025-07-22 12:07:07 -05:00
parent bd2b536ab4
commit 13ad115ca3
1 changed files with 7 additions and 0 deletions
--- a/python/src/py_nvbench.cpp
+++ b/python/src/py_nvbench.cpp
@@ -299,6 +299,13 @@ PYBIND11_MODULE(_nvbench, m)
      return std::ref(self);
    },
    py::return_value_policy::reference);
+  py_benchmark_cls.def(
+    "set_run_once",
+    [](nvbench::benchmark_base &self, bool v) {
+      self.set_run_once(v);
+      return std::ref(self);
+    },
+    py::return_value_policy::reference);

  // == STEP 5
  // Define PyState class