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 10:45:07 +02:00
parent 0ff47ad6ba
commit b20a2e2902
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/utils.h
+1 -1
View File
@@ -21,7 +21,7 @@
#define TIMER_STOP clock_gettime(CLOCK_MONOTONIC, &end); \
double t_ns = ((double)(end.tv_sec - start.tv_sec) * 1.0e9 + \
(double)(end.tv_nsec - start.tv_nsec))/1.0e6; \
std::cout<<"Time:"<<std::setw(16)<<t_ns<<" ms\n";
std::cout<<"\033[1;36mTime:"<<std::setw(16)<<t_ns<<" ms\033[0m\n";
/********************************************************
tests/yolo/yolo.cpp
+3 -4
View File
@@ -129,9 +129,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 Diffs: "<<checkResult(out_dim, out_data, out)<<"\n";
std::cout<<"TRT vs correct Diffs: "<<checkResult(out_dim, out_data2, out)<<"\n";
std::cout<<"CUDNN vs TRT Diffs: "<<checkResult(out_dim, out_data, out)<<"\n";
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";
return 0;
}