summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--contrib/lua-torch/torch7/lib/TH/THGeneral.c4
1 files changed, 4 insertions, 0 deletions
diff --git a/contrib/lua-torch/torch7/lib/TH/THGeneral.c b/contrib/lua-torch/torch7/lib/TH/THGeneral.c
index 998ac3212..03a3a0c3f 100644
--- 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)
{