path: root/fastnn/init.c

#include <lua.h>
#include <lauxlib.h>

#if LUA_VERSION_NUM == 501
static void luaL_setfuncs(lua_State *L, const luaL_Reg *l, int nup)
{
  luaL_checkstack(L, nup+1, "too many upvalues");
  for (; l->name != NULL; l++) {  /* fill the table with given functions */
    int i;
    lua_pushstring(L, l->name);
    for (i = 0; i < nup; i++)  /* copy upvalues to the top */
      lua_pushvalue(L, -(nup+1));
    lua_pushcclosure(L, l->func, nup);  /* closure with those upvalues */
    lua_settable(L, -(nup + 3));
  }
  lua_pop(L, nup);  /* remove upvalues */
}
#endif


extern void fastnn_init_modelsync(lua_State *L);
extern void fastnn_init_example(lua_State *L);
extern void fastnn_init_example_repo(lua_State *L);
extern void fastnn_init_device(lua_State *L);
extern void fastnn_init_context(lua_State *L);

int luaopen_libfastnn(lua_State *L) 
{
  //printf("%s %lx\n", model_sync__[13].name, model_sync__[13].func);
  lua_newtable(L);
      /* duplicate table */
  lua_pushvalue(L, -1);
    /* set table to global index */
  lua_setfield(L, LUA_GLOBALSINDEX, "fastnn");

  fastnn_init_modelsync(L);
  fastnn_init_example(L);
  fastnn_init_example_repo(L);
  fastnn_init_device(L);
  fastnn_init_context(L);
  return 1;
}