From 2c671761a7a88321b321d2a8d7288f65466b2eb3 Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov Date: Wed, 26 Aug 2020 16:11:33 +0100 Subject: [PATCH] [Minor] Lua_tensor: Add has_blas method --- src/lua/lua_tensor.c | 14 ++++++++++++++ 1 file changed, 14 insertions(+) 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}, }; @@ -714,6 +716,18 @@ lua_tensor_transpose (lua_State *L) return 1; } +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) { -- 2.39.5