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

Fix pooling, add return code in each test, add return code handling in test_all_tests script

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-04-09 15:25:05 +02:00
parent ed0596a52c
commit 3502c5b676
31 changed files with 293 additions and 202 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/imuodom/imuodom.cpp
+4 -3
View File
@@ -51,6 +51,7 @@ int main() {
std::ofstream path("path.txt");
int ret_cudnn = 0;
for(int i=0; i<N; i++) {
std::cout<<"i: "<<i<<"\n";
//TIMER_START
@@ -65,9 +66,9 @@ int main() {
// Print real test
printCenteredTitle( (std::string(" CHECK RESULT ") + std::to_string(i) + " ").c_str() , '=');
ImuNet.odim0.print();
checkResult(ImuNet.odim0.tot(), out0, ImuNet.o0_d);
ret_cudnn |= checkResult(ImuNet.odim0.tot(), out0, ImuNet.o0_d) == 0 ? 0 : ERROR_CUDNN;
ImuNet.odim1.print();
checkResult(ImuNet.odim0.tot(), out1, ImuNet.o1_d);
ret_cudnn |= checkResult(ImuNet.odim0.tot(), out1, ImuNet.o1_d) == 0 ? 0 : ERROR_CUDNN;
i0_h += ImuNet.dim0.tot();
i1_h += ImuNet.dim1.tot();
@@ -77,5 +78,5 @@ int main() {
}
system("cat path.txt | gnuplot -p -e \"set datafile separator ' '; plot '-'\"");
return 0;
return ret_cudnn;
}