aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/lua-torch/torch7/utils.h
diff options
context:
space:
mode:
Diffstat (limited to 'contrib/lua-torch/torch7/utils.h')
-rw-r--r--contrib/lua-torch/torch7/utils.h36
1 files changed, 36 insertions, 0 deletions
diff --git a/contrib/lua-torch/torch7/utils.h b/contrib/lua-torch/torch7/utils.h
new file mode 100644
index 000000000..f9654d4b7
--- /dev/null
+++ b/contrib/lua-torch/torch7/utils.h
@@ -0,0 +1,36 @@
+#ifndef TORCH_UTILS_INC
+#define TORCH_UTILS_INC
+
+#include "luaT.h"
+#include "TH.h"
+
+#include <lua.h>
+#include <lualib.h>
+
+#ifdef _WIN32
+#else
+#include <unistd.h>
+#endif
+
+#ifdef __cplusplus
+# define TORCH_EXTERNC extern "C"
+#else
+# define TORCH_EXTERNC extern
+#endif
+
+#ifdef _WIN32
+# ifdef torch_EXPORTS
+# define TORCH_API TORCH_EXTERNC __declspec(dllexport)
+# else
+# define TORCH_API TORCH_EXTERNC __declspec(dllimport)
+# endif
+#else
+# define TORCH_API TORCH_EXTERNC
+#endif
+
+
+TORCH_API THLongStorage* torch_checklongargs(lua_State *L, int index);
+TORCH_API int torch_islongargs(lua_State *L, int index);
+TORCH_API const char* torch_getdefaulttensortype(lua_State *L);
+
+#endif