未验证 提交 16ec22c4 编写于 作者: Y Yuanle Liu 提交者: GitHub

adjust binding order (#52225)

上级 e3217e3e
...@@ -1097,8 +1097,9 @@ void BindZeroCopyTensor(py::module *m) { ...@@ -1097,8 +1097,9 @@ void BindZeroCopyTensor(py::module *m) {
.def("copy_from_cpu", &ZeroCopyTensorCreate<uint8_t>) .def("copy_from_cpu", &ZeroCopyTensorCreate<uint8_t>)
.def("copy_from_cpu", &ZeroCopyTensorCreate<int32_t>) .def("copy_from_cpu", &ZeroCopyTensorCreate<int32_t>)
.def("copy_from_cpu", &ZeroCopyTensorCreate<int64_t>) .def("copy_from_cpu", &ZeroCopyTensorCreate<int64_t>)
.def("copy_from_cpu", &ZeroCopyTensorCreate<double>)
.def("copy_from_cpu", &ZeroCopyTensorCreate<float>) .def("copy_from_cpu", &ZeroCopyTensorCreate<float>)
// NOTE(liuyuanle): double must be bound after float.
.def("copy_from_cpu", &ZeroCopyTensorCreate<double>)
.def("copy_from_cpu", &ZeroCopyTensorCreate<paddle_infer::float16>) .def("copy_from_cpu", &ZeroCopyTensorCreate<paddle_infer::float16>)
.def("copy_from_cpu", &ZeroCopyTensorCreate<bool>) .def("copy_from_cpu", &ZeroCopyTensorCreate<bool>)
.def("copy_from_cpu", &ZeroCopyStringTensorCreate) .def("copy_from_cpu", &ZeroCopyStringTensorCreate)
...@@ -1121,8 +1122,9 @@ void BindPaddleInferTensor(py::module *m) { ...@@ -1121,8 +1122,9 @@ void BindPaddleInferTensor(py::module *m) {
.def("_copy_from_cpu_bind", &PaddleInferTensorCreate<uint8_t>) .def("_copy_from_cpu_bind", &PaddleInferTensorCreate<uint8_t>)
.def("_copy_from_cpu_bind", &PaddleInferTensorCreate<int32_t>) .def("_copy_from_cpu_bind", &PaddleInferTensorCreate<int32_t>)
.def("_copy_from_cpu_bind", &PaddleInferTensorCreate<int64_t>) .def("_copy_from_cpu_bind", &PaddleInferTensorCreate<int64_t>)
.def("_copy_from_cpu_bind", &PaddleInferTensorCreate<double>)
.def("_copy_from_cpu_bind", &PaddleInferTensorCreate<float>) .def("_copy_from_cpu_bind", &PaddleInferTensorCreate<float>)
// NOTE(liuyuanle): double must be bound after float.
.def("_copy_from_cpu_bind", &PaddleInferTensorCreate<double>)
.def("_copy_from_cpu_bind", .def("_copy_from_cpu_bind",
&PaddleInferTensorCreate<paddle_infer::float16>) &PaddleInferTensorCreate<paddle_infer::float16>)
.def("_copy_from_cpu_bind", &PaddleInferTensorCreate<bool>) .def("_copy_from_cpu_bind", &PaddleInferTensorCreate<bool>)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册