From 714eb56e1760fdfb26afccde92664d3a2f1e8435 Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov Date: Wed, 23 May 2018 18:14:15 +0100 Subject: [Minor] Move lua contrib libraries to lua- prefix --- contrib/lua-torch/torch7/random.lua | 53 +++++++++++++++++++++++++++++++++++++ 1 file changed, 53 insertions(+) create mode 100644 contrib/lua-torch/torch7/random.lua (limited to 'contrib/lua-torch/torch7/random.lua') diff --git a/contrib/lua-torch/torch7/random.lua b/contrib/lua-torch/torch7/random.lua new file mode 100644 index 000000000..59bbd7b1a --- /dev/null +++ b/contrib/lua-torch/torch7/random.lua @@ -0,0 +1,53 @@ +local wrap = require 'cwrap' + +require 'torchcwrap' + +local interface = wrap.CInterface.new() + +interface:print( + [[ +#include "luaT.h" +#include "TH.h" + +extern void torch_Generator_init(lua_State *L); +extern void torch_Generator_new(lua_State *L); + ]]) + +for _,name in ipairs({"seed", "initialSeed"}) do + interface:wrap(name, + string.format("THRandom_%s",name), + {{name='Generator', default=true}, + {name="long", creturned=true}}) +end + +interface:wrap('manualSeed', + 'THRandom_manualSeed', + {{name='Generator', default=true}, + {name="long"}}) + +interface:wrap('getRNGState', + 'THByteTensor_getRNGState', + {{name='Generator', default=true}, + {name='ByteTensor',default=true,returned=true,method={default='nil'}} + }) + +interface:wrap('setRNGState', + 'THByteTensor_setRNGState', + {{name='Generator', default=true}, + {name='ByteTensor',default=true,returned=true,method={default='nil'}} + }) + +interface:register("random__") + +interface:print( + [[ +void torch_random_init(lua_State *L) +{ + torch_Generator_init(L); + torch_Generator_new(L); + lua_setfield(L, -2, "_gen"); + luaT_setfuncs(L, random__, 0); +} +]]) + +interface:tofile(arg[1]) -- cgit v1.2.3