Skip to content

Commit 275f8ee

Browse files
Implement Benchmark.setName
1 parent c73df31 commit 275f8ee

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

python/src/py_nvbench.cpp

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)