Searched refs:my_func (Results 1 - 1 of 1) sorted by relevance

/gem5/ext/pybind11/tests/
H A Dtest_numpy_vectorize.cpp14 double my_func(int x, float y, double z) { function
15 py::print("my_func(x:int={}, y:float={:.0f}, z:float={:.0f})"_s.format(x, y, z));
25 m.def("vectorized_func", py::vectorize(my_func));
30 return py::vectorize([z](int x, float y) { return my_func(x, y, z); })(x, y);

Completed in 3 milliseconds