@@ -246,7 +246,7 @@ PYBIND11_MODULE(_nvbench, m)
246246 py_benchmark_cls.def (
247247 " addInt64Axis" ,
248248 [](nvbench::benchmark_base &self, std::string name, const std::vector<nvbench::int64_t > &data) {
249- self.add_int64_axis (name, data);
249+ self.add_int64_axis (std::move ( name) , data);
250250 return std::ref (self);
251251 },
252252 py::return_value_policy::reference);
@@ -255,14 +255,21 @@ PYBIND11_MODULE(_nvbench, m)
255255 [](nvbench::benchmark_base &self,
256256 std::string name,
257257 const std::vector<nvbench::float64_t > &data) {
258- self.add_float64_axis (name, data);
258+ self.add_float64_axis (std::move ( name) , data);
259259 return std::ref (self);
260260 },
261261 py::return_value_policy::reference);
262262 py_benchmark_cls.def (
263263 " addStringAxis" ,
264264 [](nvbench::benchmark_base &self, std::string name, const std::vector<std::string> &data) {
265- self.add_string_axis (name, data);
265+ self.add_string_axis (std::move (name), data);
266+ return std::ref (self);
267+ },
268+ py::return_value_policy::reference);
269+ py_benchmark_cls.def (
270+ " setName" ,
271+ [](nvbench::benchmark_base &self, std::string name) {
272+ self.set_name (std::move (name));
266273 return std::ref (self);
267274 },
268275 py::return_value_policy::reference);
0 commit comments