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

[Bug Fix] fix pre-commit error (#471)

Browse Source 
11-01/10:16
* 修复commit错误
This commit is contained in:
Zheng_Bicheng
2022-11-01 16:29:31 +08:00
committed by GitHub
parent 4ffcfbe726
commit bc089f47be
3 changed files with 23 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
fastdeploy/backends/rknpu/rknpu2/rknpu2_backend.cc
View File

@@ -15,10 +15,10 @@
namespace fastdeploy {
RKNPU2Backend::~RKNPU2Backend() {
if(input_attrs != nullptr){
if (input_attrs != nullptr) {
free(input_attrs);
}
if(output_attrs != nullptr){
if (output_attrs != nullptr) {
free(output_attrs);
}
}
@@ -150,7 +150,8 @@ bool RKNPU2Backend::GetModelInputOutputInfos() {
}
// Get detailed input parameters
input_attrs = (rknn_tensor_attr*)malloc(sizeof(rknn_tensor_attr) * io_num.n_input);
input_attrs =
(rknn_tensor_attr*)malloc(sizeof(rknn_tensor_attr) * io_num.n_input);
memset(input_attrs, 0, io_num.n_input * sizeof(rknn_tensor_attr));
inputs_desc_.resize(io_num.n_input);
for (uint32_t i = 0; i < io_num.n_input; i++) {
@@ -348,7 +349,8 @@ bool RKNPU2Backend::Infer(std::vector<FDTensor>& inputs,
}
(*outputs)[i].Resize(temp_shape, outputs_desc_[i].dtype,
outputs_desc_[i].name);
memcpy((*outputs)[i].MutableData(), (float*)output_mems[i]->virt_addr, (*outputs)[i].Nbytes());
memcpy((*outputs)[i].MutableData(), (float*)output_mems[i]->virt_addr,
(*outputs)[i].Nbytes());
rknn_destroy_mem(ctx, output_mems[i]);
}