mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

dealloc in test.h

Browse Source 
fix #36
This commit is contained in:
Francesco Gatti
2020-06-16 12:44:12 +02:00
parent f9c50b2a49
commit 517eaa553b
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/tkDNN/test.h
+8 -4
View File
@@ -1,7 +1,7 @@
#include <tkdnn.h> #include <tkdnn.h>
int testInference(std::vector<std::string> input_bins, std::vector<std::string> output_bins, int testInference(std::vector<std::string> input_bins, std::vector<std::string> output_bins,
tk::dnn::Network *net, tk::dnn::NetworkRT *netRT = nullptr) { tk::dnn::Network *net, tk::dnn::NetworkRT *netRT = nullptr) {
std::vector<tk::dnn::Layer*> outputs; std::vector<tk::dnn::Layer*> outputs;
for(int i=0; i<net->num_layers; i++) { for(int i=0; i<net->num_layers; i++) {
@@ -67,7 +67,11 @@ int testInference(std::vector<std::string> input_bins, std::vector<std::string>
std::cout<<"CUDNN vs TRT "; std::cout<<"CUDNN vs TRT ";
ret_cudnn_tensorrt |= checkResult(odim, cudnn_out[i], rt_out[i]) == 0 ? 0 : ERROR_CUDNNvsTENSORRT; ret_cudnn_tensorrt |= checkResult(odim, cudnn_out[i], rt_out[i]) == 0 ? 0 : ERROR_CUDNNvsTENSORRT;
} }
}
return ret_cudnn | ret_tensorrt | ret_cudnn_tensorrt;
} delete [] out_h;
checkCuda( cudaFree(out) );
}
delete [] input_h;
checkCuda( cudaFree(data) );
return ret_cudnn | ret_tensorrt | ret_cudnn_tensorrt;
}