add double precision matrix implementation

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);
 }