[Minor] Make Torch less dumb, sigh

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