From d07db44bcb04eb177d9ad289694d8ebe5daa9a4b Mon Sep 17 00:00:00 2001
From: ioir123ju <549981178@qq.com>
Date: Wed, 17 Jun 2020 16:58:57 +0800
Subject: [PATCH] [CHG] ADD xcalloc

---
 include/tkDNN/utils.h | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/include/tkDNN/utils.h b/include/tkDNN/utils.h
index aa73e9e..d419920 100644
--- a/include/tkDNN/utils.h
+++ b/include/tkDNN/utils.h
@@ -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);