diff options
Diffstat (limited to 'contrib/lua-torch')
| -rw-r--r-- | contrib/lua-torch/torch7/lib/TH/THGeneral.c | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/contrib/lua-torch/torch7/lib/TH/THGeneral.c b/contrib/lua-torch/torch7/lib/TH/THGeneral.c index 03a3a0c3f..f093c422f 100644 --- 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) { |