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

fix

Browse Source
This commit is contained in:
Francesco Gatti
2017-08-04 16:16:03 +00:00
parent b20a2e2902
commit 34198a4e8d
6 changed files with 55 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/mnist/test_mnist.cpp
+1 -1
View File
@@ -58,7 +58,7 @@ int main() {
//printDeviceVector(10, out_data);
std::cout<<"\n======= CHECK RESULT =======\n";
std::cout<<"Diffs: "<<checkResult(dim.tot(), out_data, out_data2)<<"\n";
checkResult(dim.tot(), out_data, out_data2);
/*
// Print real test
tests/yolo/yolo.cpp
+7 -6
View File
@@ -103,13 +103,14 @@ int main() {
value_type *input_h;
readBinaryFile(input_bin, dim.tot(), &input_h, &data);
//convert network to tensorRT
tkDNN::NetworkRT netRT(&net);
value_type *out_data, *out_data2;
value_type *out_data, *out_data2; // cudnn output, tensorRT output
tkDNN::dataDim_t dim1 = dim;
tkDNN::dataDim_t dim1 = dim; //input dim
std::cout<<"\n==== CUDNN inference =======\n"; {
dim1.print(); //print initial dimension
dim1.print();
TIMER_START
out_data = net.infer(dim1, data);
TIMER_STOP
@@ -129,8 +130,8 @@ int main() {
value_type *out, *out_h;
int out_dim = net.getOutputDim().tot();
readBinaryFile(output_bin, out_dim, &out_h, &out);
std::cout<<"CUDNN vs correct Wrongs: "<<checkResult(out_dim, out_data, out)<<"\n";
std::cout<<"TRT vs correct Wrongs: "<<checkResult(out_dim, out_data2, out)<<"\n";
std::cout<<"CUDNN vs TRT Wrongs: "<<checkResult(out_dim, out_data, out)<<"\n";
std::cout<<"CUDNN vs correct"; checkResult(out_dim, out_data, out);
std::cout<<"TRT vs correct"; checkResult(out_dim, out_data2, out);
std::cout<<"CUDNN vs TRT "; checkResult(out_dim, out_data, out_data2);
return 0;
}