expose storage and tensor retain() and free() methods into Lua

2 files changed, 18 insertions, 0 deletions

diff --git a/generic/Storage.c b/generic/Storage.c
index 66ff742..9dc3ba2 100644
--- a/generic/Storage.c
+++ b/generic/Storage.c
@@ -45,6 +45,13 @@ static int torch_Storage_(new)(lua_State *L)
   return 1;
 }
 
+static int torch_Storage_(retain)(lua_State *L)
+{
+  THStorage *storage = luaT_checkudata(L, 1, torch_Storage);
+  THStorage_(retain)(storage);
+  return 0;
+}
+
 static int torch_Storage_(free)(lua_State *L)
 {
   THStorage *storage = luaT_checkudata(L, 1, torch_Storage);
@@ -201,6 +208,8 @@ static int torch_Storage_(read)(lua_State *L)
 }
 
 static const struct luaL_Reg torch_Storage_(_) [] = {
+  {"retain", torch_Storage_(retain)},
+  {"free", torch_Storage_(free)},
   {"size", torch_Storage_(__len__)},
   {"__len__", torch_Storage_(__len__)},
   {"__newindex__", torch_Storage_(__newindex__)},
diff --git a/generic/Tensor.c b/generic/Tensor.c
index 730d4a9..c8f84d2 100644
--- a/generic/Tensor.c
+++ b/generic/Tensor.c
@@ -956,6 +956,13 @@ static int torch_Tensor_(__index__)(lua_State *L)
   }
 }
 
+static int torch_Tensor_(retain)(lua_State *L)
+{
+  THTensor *tensor = luaT_checkudata(L, 1, torch_Tensor);
+  THTensor_(retain)(tensor);
+  return 0;
+}
+
 static int torch_Tensor_(free)(lua_State *L)
 {
   THTensor *tensor = luaT_checkudata(L, 1, torch_Tensor);
@@ -1217,6 +1224,8 @@ static int torch_Tensor_(read)(lua_State *L)
 }
 
 static const struct luaL_Reg torch_Tensor_(_) [] = {
+  {"retain", torch_Tensor_(retain)},
+  {"free", torch_Tensor_(free)},
   {"contiguous", torch_Tensor_(contiguous)},
   {"size", torch_Tensor_(size)},
   {"__len__", torch_Tensor_(size)},