Provide stub for avx512 vnni architecture

Author: serge-sans-paille

include/xsimd/arch/xsimd_isa.hpp

@@ -84,6 +84,10 @@
 #include "./xsimd_avx512vbmi.hpp"
 #endif
 
+#if XSIMD_WITH_AVX512VNNI
+#include "./xsimd_avx512vnni.hpp"
+#endif
+
 #if XSIMD_WITH_NEON
 #include "./xsimd_neon.hpp"
 #endif

include/xsimd/config/xsimd_config.hpp

@@ -288,6 +288,17 @@
 #define XSIMD_WITH_AVX512VBMI 0
 #endif
 
+/**
+ * @ingroup xsimd_config_macro
+ *
+ * Set to 1 if AVX512VNNI is available at compile-time, to 0 otherwise.
+ */
+#ifdef __AVX512VNNI__
+#define XSIMD_WITH_AVX512VNNI XSIMD_WITH_AVX512F
+#else
+#define XSIMD_WITH_AVX512VNNI 0
+#endif
+
 #ifdef __ARM_NEON
 
 /**

include/xsimd/config/xsimd_cpuid.hpp

@@ -54,6 +54,7 @@ namespace xsimd
             unsigned avx512pf : 1;
             unsigned avx512ifma : 1;
             unsigned avx512vbmi : 1;
+            unsigned avx512vnni : 1;
             unsigned neon : 1;
             unsigned neon64 : 1;
             unsigned sve : 1;
@@ -206,6 +207,9 @@ namespace xsimd
 
                 avx512vbmi = regs7[2] >> 1 & 1;
                 best = std::max(best, avx512vbmi::version() * avx512vbmi * avx512ifma * avx512bw * avx512dq * avx512cd * avx512f);
+
+                avx512vnni = regs7[2] >> 11 & 1;
+                best = std::max(best, avx512vnni::version() * avx512vnni * avx512vbmi * avx512ifma * avx512bw * avx512dq * avx512cd * avx512f);
 #endif
             }
         };

include/xsimd/types/xsimd_all_registers.hpp

@@ -21,6 +21,8 @@
 #include "xsimd_fma3_avx2_register.hpp"
 #include "xsimd_fma3_avx_register.hpp"
 
+#include "xsimd_avx512vnni_register.hpp"
+
 #include "xsimd_avx512ifma_register.hpp"
 #include "xsimd_avx512vbmi_register.hpp"
 

include/xsimd/types/xsimd_avx512vnni_register.hpp

@@ -0,0 +1,48 @@
+/***************************************************************************
+ * Copyright (c) Johan Mabille, Sylvain Corlay, Wolf Vollprecht and         *
+ * Martin Renou                                                             *
+ * Copyright (c) QuantStack                                                 *
+ * Copyright (c) Serge Guelton                                              *
+ *                                                                          *
+ * Distributed under the terms of the BSD 3-Clause License.                 *
+ *                                                                          *
+ * The full license is in the file LICENSE, distributed with this software. *
+ ****************************************************************************/
+
+#ifndef XSIMD_AVX512VNNI_REGISTER_HPP
+#define XSIMD_AVX512VNNI_REGISTER_HPP
+
+#include "./xsimd_avx512vbmi_register.hpp"
+
+namespace xsimd
+{
+
+    /**
+     * @ingroup architectures
+     *
+     * AVX512VNNI instructions
+     */
+    struct avx512vnni : avx512vbmi
+    {
+        static constexpr bool supported() noexcept { return XSIMD_WITH_AVX512VNNI; }
+        static constexpr bool available() noexcept { return true; }
+        static constexpr unsigned version() noexcept { return generic::version(3, 6, 0); }
+        static constexpr char const* name() noexcept { return "avx512vnni"; }
+    };
+
+#if XSIMD_WITH_AVX512VNNI
+
+    namespace types
+    {
+        template <class T>
+        struct get_bool_simd_register<T, avx512vnni>
+        {
+            using type = simd_avx512_bool_register<T>;
+        };
+
+        XSIMD_DECLARE_SIMD_REGISTER_ALIAS(avx512vnni, avx512vbmi);
+
+    }
+#endif
+}
+#endif