apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-15 13:10:55 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

[FDTensor] Add device id to output tensor (#978)

Browse Source 
Add device id
This commit is contained in:
Jack Zhou
2022-12-27 16:02:57 +08:00
committed by GitHub
parent 7ce999f08e
commit 3c38ba4415
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fastdeploy/runtime.cc
View File

@@ -660,7 +660,11 @@ bool Runtime::Infer(std::vector<FDTensor>& input_tensors,
}
bool Runtime::Infer() {
return backend_->Infer(input_tensors_, &output_tensors_, false);
bool result = backend_->Infer(input_tensors_, &output_tensors_, false);
for (auto& tensor : output_tensors_) {
tensor.device_id = option.device_id;
}
return result;
}
void Runtime::BindInputTensor(const std::string& name, FDTensor& input) {