| local wrap = require 'cwrap' |
| |
| require 'torchcwrap' |
| |
| local interface = wrap.CInterface.new() |
| |
| interface:print( |
| [[ |
| #include "luaT.h" |
| #include "TH.h" |
| ]]) |
| |
| for _,name in ipairs({"seed", "initialSeed"}) do |
| interface:wrap(name, |
| string.format("THRandom_%s",name), |
| {{name="long", creturned=true}}) |
| end |
| |
| interface:wrap('manualSeed', |
| 'THRandom_manualSeed', |
| {{name="long"}}) |
| |
| interface:wrap('getRNGState','THLongTensor_getRNGState',{{name='LongTensor',default=true,returned=true,method={default='nil'}}}) |
| interface:wrap('setRNGState','THLongTensor_setRNGState',{{name='LongTensor',default=true,returned=true,method={default='nil'}}}) |
| |
| interface:register("random__") |
| |
| interface:print( |
| [[ |
| void torch_random_init(lua_State *L) |
| { |
| luaL_register(L, NULL, random__); |
| } |
| ]]) |
| |
| interface:tofile(arg[1]) |