apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-17 14:11:14 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[Runtime]FDTensor pybind add from_dlpack interface (#1001)

Browse Source 
add from_dlpack
This commit is contained in:
heliqi
2022-12-28 21:58:32 +08:00
committed by GitHub
parent 02e2a5365b
commit ab0929662b
2 changed files with 71 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
fastdeploy/pybind/runtime.cc
View File

@@ -214,7 +214,8 @@ void BindRuntime(pybind11::module& m) {
.def("infer",
[](Runtime& self, std::vector<FDTensor>& inputs) {
std::vector<FDTensor> outputs;
return self.Infer(inputs, &outputs);
self.Infer(inputs, &outputs);
return outputs;
})
.def("bind_input_tensor", &Runtime::BindInputTensor)
.def("infer", [](Runtime& self) { self.Infer(); })