diff options
Diffstat (limited to 'matrix/init.c')
| -rw-r--r-- | matrix/init.c | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/matrix/init.c b/matrix/init.c index e251628..e723f55 100644 --- a/matrix/init.c +++ b/matrix/init.c @@ -2,6 +2,8 @@ #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); static const luaL_Reg matrix_methods[] = { {"__tostring__", nerv_error_method_not_implemented }, {"__add__", nerv_error_method_not_implemented }, @@ -16,4 +18,5 @@ void nerv_matrix_init(lua_State *L) { luaL_register(L, NULL, matrix_methods); lua_pop(L, 1); nerv_float_matrix_host_init(L); + nerv_float_matrix_cuda_init(L); } |