[Minor] Fix FreeBSD build

diff --git a/contrib/lua-torch/torch7/lib/TH/THGeneral.c b/contrib/lua-torch/torch7/lib/TH/THGeneral.c
index 03a3a0c3f7b5bab3ece422709299a3df530896b8..f093c422f1a76802ec53437ed2eec7fa936b6215 100644 (file)
--- a/contrib/lua-torch/torch7/lib/TH/THGeneral.c
+++ b/contrib/lua-torch/torch7/lib/TH/THGeneral.c
@@ -15,10 +15,14 @@
 #include <malloc/malloc.h>
 #endif
 
-#ifdef __linux__
+#if defined(__linux__)
 #include <malloc.h>
 #endif
 
+#if defined(__FreeBSD__)
+#include <malloc_np.h>
+#endif
+
 /* Torch Error Handling */
 static void defaultErrorHandlerFunction(const char *msg, void *data)
 {