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

Add mnist download, change ERROR_TKDNN into ERROR_TENSORRT

Browse Source 
Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
This commit is contained in:
Micaela Verucchi
2020-04-09 19:02:25 +02:00
parent 4361d5fec0
commit fabad7bace
28 changed files with 36 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/simple/test_simple.cpp
+1 -1
View File
@@ -65,7 +65,7 @@ int main() {
// std::cout<<"CUDNN vs correct";
// int ret_cudnn = checkResult(out_dim, out_data, out) == 0 ? 0: ERROR_CUDNN;
// std::cout<<"TRT vs correct";
// int ret_tensorrt = checkResult(out_dim, out_data2, out) == 0 ? 0 : ERROR_TKDNN;
// int ret_tensorrt = checkResult(out_dim, out_data2, out) == 0 ? 0 : ERROR_TENSORRT;
std::cout<<"CUDNN vs TRT ";
int ret_cudnn_tensorrt = checkResult(out_dim, out_data, out_data2) == 0 ? 0 : ERROR_CUDNNvsTENSORRT;