From 25506e740e4c324d0a3cf6655c851cfa71de1430 Mon Sep 17 00:00:00 2001 From: Christian <12550267+iceychris@users.noreply.github.com> Date: Sat, 23 Mar 2024 17:38:54 +0100 Subject: [PATCH] Expose Tensor.setRawData in Python bindings --- python/src/main.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/src/main.cpp b/python/src/main.cpp index 6c0f640c..9a27f9d3 100644 --- a/python/src/main.cpp +++ b/python/src/main.cpp @@ -160,7 +160,11 @@ PYBIND11_MODULE(kp, m) .def("tensor_type", &kp::Tensor::tensorType, DOC(kp, Tensor, tensorType)) .def("data_type", &kp::Tensor::dataType, DOC(kp, Tensor, dataType)) .def("is_init", &kp::Tensor::isInit, DOC(kp, Tensor, isInit)) - .def("destroy", &kp::Tensor::destroy, DOC(kp, Tensor, destroy)); + .def("destroy", &kp::Tensor::destroy, DOC(kp, Tensor, destroy)) + .def("set_raw_data", [](kp::Tensor& self, py::buffer b) { + py::buffer_info info = b.request(); + self.setRawData(info.ptr); + }, DOC(kp, Tensor, setRawData)); py::class_>(m, "Sequence") .def(