dealloc in test.h

fix #36
2020-06-16 12:44:12 +02:00
parent 1b8f45703f
commit 07193fc343
1 changed files with 8 additions and 4 deletions
@@ -67,7 +67,11 @@ int testInference(std::vector<std::string> input_bins, std::vector<std::string>
            std::cout<<"CUDNN vs TRT    "; 
            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;
+}