diff options
-rw-r--r-- | contrib/lua-torch/torch7/lib/TH/THGeneral.c | 4 |
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) { |