diff options
Diffstat (limited to 'src/lua')
| -rw-r--r-- | src/lua/lua_tensor.c | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/src/lua/lua_tensor.c b/src/lua/lua_tensor.c index b40ee6e71..e918188eb 100644 --- a/src/lua/lua_tensor.c +++ b/src/lua/lua_tensor.c @@ -38,11 +38,13 @@ LUA_FUNCTION_DEF (tensor, len); LUA_FUNCTION_DEF (tensor, eigen); LUA_FUNCTION_DEF (tensor, mean); LUA_FUNCTION_DEF (tensor, transpose); +LUA_FUNCTION_DEF (tensor, has_blas); static luaL_reg rspamd_tensor_f[] = { LUA_INTERFACE_DEF (tensor, load), LUA_INTERFACE_DEF (tensor, new), LUA_INTERFACE_DEF (tensor, fromtable), + LUA_INTERFACE_DEF (tensor, has_blas), {NULL, NULL}, }; @@ -715,6 +717,18 @@ lua_tensor_transpose (lua_State *L) } static gint +lua_tensor_has_blas (lua_State *L) +{ +#ifdef HAVE_CBLAS + lua_pushboolean (L, true); +#else + lua_pushboolean (L, false); +#endif + + return 1; +} + +static gint lua_load_tensor (lua_State * L) { lua_newtable (L); |