From 7dde66a718eb8d4a7b8d1e686218a8f8b743db98 Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Sun, 10 Dec 2023 14:11:21 +0100
Subject: [PATCH] Fixed PyBind
---
python_binding/data/pybind_Tensor.cpp | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/python_binding/data/pybind_Tensor.cpp b/python_binding/data/pybind_Tensor.cpp
index 067ad8c00..fa109a9af 100644
--- a/python_binding/data/pybind_Tensor.cpp
+++ b/python_binding/data/pybind_Tensor.cpp
@@ -71,7 +71,8 @@ void init_Tensor(py::module& m){
(m,"Tensor", py::multiple_inheritance(), py::buffer_protocol());
pyClassTensor.def(py::init<>())
- .def("set_backend", &Tensor::setBackend, py::arg("name"), py::arg("device") = 0)
+ .def("set_datatype", &Tensor::setDataType, py::arg("datatype"), py::arg("copyCast") = true)
+ .def("set_backend", &Tensor::setBackend, py::arg("name"), py::arg("device") = 0, py::arg("copyFrom") = true)
.def("dims", (const std::vector<DimSize_t>& (Tensor::*)()const) &Tensor::dims)
.def("dtype", &Tensor::dataType)
.def("size", &Tensor::size)
--
GitLab