1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
#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;
}
|