From 73e895b8ce4bb8488ef4527c34498fdda95953c1 Mon Sep 17 00:00:00 2001 From: YdrMaster Date: Tue, 21 Mar 2023 14:40:13 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E5=AF=BC=E5=87=BA=E6=8B=B7=E5=87=BA?= =?UTF-8?q?=E5=BC=A0=E9=87=8F=E5=80=BC=E6=96=B9=E6=B3=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: YdrMaster --- src/ffi/ffi_infinitensor.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/ffi/ffi_infinitensor.cc b/src/ffi/ffi_infinitensor.cc index 641c2c3b..390ecb2d 100644 --- a/src/ffi/ffi_infinitensor.cc +++ b/src/ffi/ffi_infinitensor.cc @@ -178,6 +178,8 @@ void init_graph_builder(py::module &m) { .def("fuid", &TensorObj::getFuid, policy::automatic) .def("shape", &TensorObj::getDims, policy::move) .def("copyoutFloat", &TensorObj::copyoutFloat, policy::move) + .def("copyoutInt32", &TensorObj::copyoutInt32, policy::move) + .def("copyoutInt64", &TensorObj::copyoutInt64, policy::move) .def("has_target", &TensorObj::hasTarget, policy::automatic) .def("src", &TensorObj::getOutputOf, policy::move); py::class_>(m, "Operator")