diff options
Diffstat (limited to 'matrix/init.c')
| -rw-r--r-- | matrix/init.c | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/matrix/init.c b/matrix/init.c index e723f55..fb1c287 100644 --- a/matrix/init.c +++ b/matrix/init.c @@ -2,8 +2,11 @@ #include "generic/matrix.h" const char *nerv_matrix_tname = "nerv.Matrix"; -void nerv_float_matrix_host_init(lua_State *L); -void nerv_float_matrix_cuda_init(lua_State *L); +void nerv_matrix_float_host_init(lua_State *L); +void nerv_matrix_float_cuda_init(lua_State *L); +void nerv_matrix_double_host_init(lua_State *L); +void nerv_matrix_double_cuda_init(lua_State *L); + static const luaL_Reg matrix_methods[] = { {"__tostring__", nerv_error_method_not_implemented }, {"__add__", nerv_error_method_not_implemented }, @@ -17,6 +20,8 @@ void nerv_matrix_init(lua_State *L) { luaT_newmetatable(L, nerv_matrix_tname, NULL, NULL, NULL, NULL); luaL_register(L, NULL, matrix_methods); lua_pop(L, 1); - nerv_float_matrix_host_init(L); - nerv_float_matrix_cuda_init(L); + nerv_matrix_float_host_init(L); + nerv_matrix_float_cuda_init(L); +/* nerv_matrix_double_host_init(L); */ + nerv_matrix_double_cuda_init(L); } |