From 02ad6e5490a2e5f4dc7aa6ae0f25fb84bf27a8cc Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com>
Date: Thu, 3 Jul 2025 11:53:17 -0500
Subject: [PATCH] Implement Benchmark.setName

---
 python/src/py_nvbench.cpp | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/python/src/py_nvbench.cpp b/python/src/py_nvbench.cpp
index 24d5748..3551ff3 100644
--- a/python/src/py_nvbench.cpp
+++ b/python/src/py_nvbench.cpp
@@ -246,7 +246,7 @@ PYBIND11_MODULE(_nvbench, m)
   py_benchmark_cls.def(
     "addInt64Axis",
     [](nvbench::benchmark_base &self, std::string name, const std::vector<nvbench::int64_t> &data) {
-      self.add_int64_axis(name, data);
+      self.add_int64_axis(std::move(name), data);
       return std::ref(self);
     },
     py::return_value_policy::reference);
@@ -255,14 +255,21 @@ PYBIND11_MODULE(_nvbench, m)
     [](nvbench::benchmark_base &self,
        std::string name,
        const std::vector<nvbench::float64_t> &data) {
-      self.add_float64_axis(name, data);
+      self.add_float64_axis(std::move(name), data);
       return std::ref(self);
     },
     py::return_value_policy::reference);
   py_benchmark_cls.def(
     "addStringAxis",
     [](nvbench::benchmark_base &self, std::string name, const std::vector<std::string> &data) {
-      self.add_string_axis(name, data);
+      self.add_string_axis(std::move(name), data);
+      return std::ref(self);
+    },
+    py::return_value_policy::reference);
+  py_benchmark_cls.def(
+    "setName",
+    [](nvbench::benchmark_base &self, std::string name) {
+      self.set_name(std::move(name));
       return std::ref(self);
     },
     py::return_value_policy::reference);