Commit ed0625e7 authored by Paul's avatar Paul
Browse files

Formatting

parent 754dbbd4
...@@ -103,10 +103,11 @@ PYBIND11_MODULE(migraphx, m) ...@@ -103,10 +103,11 @@ PYBIND11_MODULE(migraphx, m)
py::class_<migraphx::argument>(m, "argument", py::buffer_protocol()) py::class_<migraphx::argument>(m, "argument", py::buffer_protocol())
.def_buffer([](migraphx::argument& x) -> py::buffer_info { return to_buffer_info(x); }) .def_buffer([](migraphx::argument& x) -> py::buffer_info { return to_buffer_info(x); })
.def("__init__", [](migraphx::argument& x, py::buffer b) { .def("__init__",
py::buffer_info info = b.request(); [](migraphx::argument& x, py::buffer b) {
new(&x) migraphx::argument(to_shape(info), info.ptr); py::buffer_info info = b.request();
}) new(&x) migraphx::argument(to_shape(info), info.ptr);
})
.def("__eq__", std::equal_to<migraphx::argument>{}) .def("__eq__", std::equal_to<migraphx::argument>{})
.def("__ne__", std::not_equal_to<migraphx::argument>{}) .def("__ne__", std::not_equal_to<migraphx::argument>{})
.def("__repr__", [](const migraphx::argument& x) { return migraphx::to_string(x); }); .def("__repr__", [](const migraphx::argument& x) { return migraphx::to_string(x); });
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment