adding getDevice for tensor, manualSeedAll and seedAll

1 files changed, 16 insertions, 0 deletions

diff --git a/init.c b/init.c
index 1f21a4c..c83ff0e 100644
--- a/init.c
+++ b/init.c
@@ -104,6 +104,13 @@ static int cutorch_seed(lua_State *L)
   return 1;
 }
 
+static int cutorch_seedAll(lua_State *L)
+{
+  unsigned long seed = THCRandom_seedAll(getState(L)->rngState);
+  lua_pushnumber(L, seed);
+  return 1;
+}
+
 static int cutorch_initialSeed(lua_State *L)
 {
   unsigned long seed = THCRandom_initialSeed(getState(L)->rngState);
@@ -118,6 +125,13 @@ static int cutorch_manualSeed(lua_State *L)
   return 0;
 }
 
+static int cutorch_manualSeedAll(lua_State* L)
+{
+  unsigned long seed = luaL_checknumber(L, 1);
+  THCRandom_manualSeedAll(getState(L)->rngState, seed);
+  return 0;
+}
+
 static int cutorch_getRNGState(lua_State *L)
 {
   THByteTensor* t = THByteTensor_new();
@@ -141,8 +155,10 @@ static const struct luaL_Reg cutorch_stuff__ [] = {
   {"getDeviceProperties", cutorch_getDeviceProperties},
   {"setDevice", cutorch_setDevice},
   {"seed", cutorch_seed},
+  {"seedAll", cutorch_seedAll},
   {"initialSeed", cutorch_initialSeed},
   {"manualSeed", cutorch_manualSeed},
+  {"manualSeedAll", cutorch_manualSeedAll},
   {"getRNGState", cutorch_getRNGState},
   {"setRNGState", cutorch_setRNGState},
   {NULL, NULL}