From 902faa66e3da52a1a4cf7b1dc7da6e6bf0d47d34 Mon Sep 17 00:00:00 2001 From: Determinant Date: Fri, 15 May 2015 03:13:31 +0800 Subject: add cumatrix --- matrix/init.c | 3 +++ 1 file changed, 3 insertions(+) (limited to 'matrix/init.c') 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); } -- cgit v1.2.3