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

Implement Benchmark.setName

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2025-07-03 11:53:17 -05:00
parent 8589511f61
commit 02ad6e5490
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
python/src/py_nvbench.cpp
View File

@@ -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);