[CHG] ADD xcalloc

2020-06-17 16:58:57 +08:00
parent 40d3018a3e
commit d07db44bcb
1 changed files with 7 additions and 1 deletions
@@ -6,7 +6,7 @@
 #include <fstream>
 #include <iomanip>
 #include <stdlib.h>
-
+#include <malloc.h>
 #include "cuda.h"
 #include "cuda_runtime_api.h"
 #include <cublas_v2.h>
@@ -101,6 +101,12 @@ typedef enum {
  ERROR_CUDNNvsTENSORRT = 8    
 } resultError_t;

+void *xmalloc(size_t size);
+void *xcalloc(size_t nmemb, size_t size);
+void malloc_error();
+void calloc_error();
+void realloc_error();
+
 void printCenteredTitle(const char *title, char fill, int dim = 30);
 bool fileExist(const char *fname);
 void downloadWeightsifDoNotExist(const std::string& input_bin, const std::string& test_folder, const std::string& weights_url);