Lines Matching refs:nb
14 namespace nb = nanobind;
38 nb::callable f) {
46 nb::enum_<TimeUnit>(m, "TimeUnit")
54 nb::enum_<BigO>(m, "BigO")
67 nb::class_<Benchmark>(m, "Benchmark")
76 .def("unit", &Benchmark::Unit, nb::rv_policy::reference)
77 .def("arg", &Benchmark::Arg, nb::rv_policy::reference)
78 .def("args", &Benchmark::Args, nb::rv_policy::reference)
79 .def("range", &Benchmark::Range, nb::rv_policy::reference,
80 nb::arg("start"), nb::arg("limit"))
82 nb::rv_policy::reference, nb::arg("start"),
83 nb::arg("limit"), nb::arg("step") = 1)
84 .def("ranges", &Benchmark::Ranges, nb::rv_policy::reference)
86 nb::rv_policy::reference)
87 .def("arg_name", &Benchmark::ArgName, nb::rv_policy::reference)
89 nb::rv_policy::reference)
91 nb::rv_policy::reference, nb::arg("lo1"), nb::arg("hi1"),
92 nb::arg("lo2"), nb::arg("hi2"))
94 nb::rv_policy::reference)
95 .def("min_time", &Benchmark::MinTime, nb::rv_policy::reference)
97 nb::rv_policy::reference)
99 nb::rv_policy::reference)
101 nb::rv_policy::reference)
103 nb::rv_policy::reference, nb::arg("value") = true)
105 nb::rv_policy::reference, nb::arg("value") = true)
107 nb::rv_policy::reference)
109 nb::rv_policy::reference)
111 nb::rv_policy::reference)
115 nb::rv_policy::reference,
116 nb::arg("complexity") = benchmark::oAuto);
119 nb::class_<Counter> py_counter(m, "Counter");
121 nb::enum_<Counter::Flags>(py_counter, "Flags")
133 .def(nb::self | nb::self);
135 nb::enum_<Counter::OneK>(py_counter, "OneK")
141 .def(nb::init<double, Counter::Flags, Counter::OneK>(),
142 nb::arg("value") = 0., nb::arg("flags") = Counter::kDefaults,
143 nb::arg("k") = Counter::kIs1000)
148 .def(nb::init_implicit<double>());
150 nb::implicitly_convertible<nb::int_, Counter>();
152 nb::bind_map<benchmark::UserCounters>(m, "UserCounters");
155 nb::class_<State>(m, "State")
170 .def("range", &State::range, nb::arg("pos") = 0)
179 nb::rv_policy::reference);