]> source.dussan.org Git - rspamd.git/commitdiff
[Minor] Lua_tensor: Add has_blas method
authorVsevolod Stakhov <vsevolod@highsecure.ru>
Wed, 26 Aug 2020 15:11:33 +0000 (16:11 +0100)
committerVsevolod Stakhov <vsevolod@highsecure.ru>
Wed, 26 Aug 2020 15:11:33 +0000 (16:11 +0100)
src/lua/lua_tensor.c

index b40ee6e7103d230ea13bd981f026de0a1939eed4..e918188ebecca727d12a2d4b8f363820338ef8d6 100644 (file)
@@ -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)
 {