Initial support for avx512vbmi

Author: junparser

include/xsimd/arch/xsimd_avx512vbmi2.hpp

@@ -0,0 +1,20 @@
+/***************************************************************************
+ * 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_AVX512VBMI2_HPP
+#define XSIMD_AVX512VBMI2_HPP
+
+#include <array>
+#include <type_traits>
+
+#include "../types/xsimd_avx512vbmi2_register.hpp"
+
+#endif

include/xsimd/arch/xsimd_isa.hpp

@@ -92,6 +92,10 @@
 #include "./xsimd_avx512vbmi.hpp"
 #endif
 
+#if XSIMD_WITH_AVX512VBMI2
+#include "./xsimd_avx512vbmi2.hpp"
+#endif
+
 #if XSIMD_WITH_AVX512VNNI_AVX512BW
 #include "./xsimd_avx512vnni_avx512bw.hpp"
 #endif

include/xsimd/config/xsimd_arch.hpp

@@ -162,7 +162,7 @@ namespace xsimd
     } // namespace detail
 
     using all_x86_architectures = arch_list<
-        avx512vnni<avx512vbmi>, avx512vbmi, avx512ifma, avx512pf, avx512vnni<avx512bw>, avx512bw, avx512er, avx512dq, avx512cd, avx512f,
+        avx512vnni<avx512vbmi>, avx512vbmi2, avx512vbmi, avx512ifma, avx512pf, avx512vnni<avx512bw>, avx512bw, avx512er, avx512dq, avx512cd, avx512f,
         avxvnni, fma3<avx2>, avx2, fma3<avx>, avx, fma4, fma3<sse4_2>,
         sse4_2, sse4_1, /*sse4a,*/ ssse3, sse3, sse2>;
 

include/xsimd/config/xsimd_config.hpp

@@ -299,6 +299,17 @@
 #define XSIMD_WITH_AVX512VBMI 0
 #endif
 
+/**
+ * @ingroup xsimd_config_macro
+ *
+ * Set to 1 if AVX512VBMI2 is available at compile-time, to 0 otherwise.
+ */
+#ifdef __AVX512VBMI2__
+#define XSIMD_WITH_AVX512VBMI2 XSIMD_WITH_AVX512F
+#else
+#define XSIMD_WITH_AVX512VBMI2 0
+#endif
+
 /**
  * @ingroup xsimd_config_macro
  *
@@ -455,7 +466,7 @@
 
 #endif
 
-#if !XSIMD_WITH_SSE2 && !XSIMD_WITH_SSE3 && !XSIMD_WITH_SSSE3 && !XSIMD_WITH_SSE4_1 && !XSIMD_WITH_SSE4_2 && !XSIMD_WITH_AVX && !XSIMD_WITH_AVX2 && !XSIMD_WITH_AVXVNNI && !XSIMD_WITH_FMA3_SSE && !XSIMD_WITH_FMA4 && !XSIMD_WITH_FMA3_AVX && !XSIMD_WITH_FMA3_AVX2 && !XSIMD_WITH_AVX512F && !XSIMD_WITH_AVX512CD && !XSIMD_WITH_AVX512DQ && !XSIMD_WITH_AVX512BW && !XSIMD_WITH_AVX512ER && !XSIMD_WITH_AVX512PF && !XSIMD_WITH_AVX512IFMA && !XSIMD_WITH_AVX512VBMI && !XSIMD_WITH_NEON && !XSIMD_WITH_NEON64 && !XSIMD_WITH_SVE && !XSIMD_WITH_RVV && !XSIMD_WITH_WASM
+#if !XSIMD_WITH_SSE2 && !XSIMD_WITH_SSE3 && !XSIMD_WITH_SSSE3 && !XSIMD_WITH_SSE4_1 && !XSIMD_WITH_SSE4_2 && !XSIMD_WITH_AVX && !XSIMD_WITH_AVX2 && !XSIMD_WITH_AVXVNNI && !XSIMD_WITH_FMA3_SSE && !XSIMD_WITH_FMA4 && !XSIMD_WITH_FMA3_AVX && !XSIMD_WITH_FMA3_AVX2 && !XSIMD_WITH_AVX512F && !XSIMD_WITH_AVX512CD && !XSIMD_WITH_AVX512DQ && !XSIMD_WITH_AVX512BW && !XSIMD_WITH_AVX512ER && !XSIMD_WITH_AVX512PF && !XSIMD_WITH_AVX512IFMA && !XSIMD_WITH_AVX512VBMI && !XSIMD_WITH_AVX512VBMI2 && !XSIMD_WITH_NEON && !XSIMD_WITH_NEON64 && !XSIMD_WITH_SVE && !XSIMD_WITH_RVV && !XSIMD_WITH_WASM
 #define XSIMD_NO_SUPPORTED_ARCHITECTURE
 #endif
 

include/xsimd/config/xsimd_cpuid.hpp

@@ -71,6 +71,7 @@ namespace xsimd
             ARCH_FIELD(avx512pf)
             ARCH_FIELD(avx512ifma)
             ARCH_FIELD(avx512vbmi)
+            ARCH_FIELD(avx512vbmi2)
             ARCH_FIELD_EX(avx512vnni<::xsimd::avx512bw>, avx512vnni_bw)
             ARCH_FIELD_EX(avx512vnni<::xsimd::avx512vbmi>, avx512vnni_vbmi)
             ARCH_FIELD(neon)
@@ -245,6 +246,7 @@ namespace xsimd
                 avx512pf = regs7[1] >> 26 & avx512_state_os_enabled;
                 avx512ifma = regs7[1] >> 21 & avx512_state_os_enabled;
                 avx512vbmi = regs7[2] >> 1 & avx512_state_os_enabled;
+                avx512vbmi2 = regs7[2] >> 6 & avx512_state_os_enabled;
                 avx512vnni_bw = regs7[2] >> 11 & avx512_state_os_enabled;
                 avx512vnni_vbmi = avx512vbmi && avx512vnni_bw;
 #endif

include/xsimd/types/xsimd_all_registers.hpp

@@ -26,6 +26,7 @@
 #include "xsimd_avx512vnni_avx512vbmi_register.hpp"
 
 #include "xsimd_avx512ifma_register.hpp"
+#include "xsimd_avx512vbmi2_register.hpp"
 #include "xsimd_avx512vbmi_register.hpp"
 
 #include "xsimd_avx512er_register.hpp"

include/xsimd/types/xsimd_avx512vbmi2_register.hpp

@@ -0,0 +1,51 @@
+/***************************************************************************
+ * 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_AVX512VBMI2_REGISTER_HPP
+#define XSIMD_AVX512VBMI2_REGISTER_HPP
+
+#include "./xsimd_avx512vbmi_register.hpp"
+
+namespace xsimd
+{
+
+    /**
+     * @ingroup architectures
+     *
+     * AVX512VBMI instructions
+     */
+    struct avx512vbmi2 : avx512vbmi
+    {
+        static constexpr bool supported() noexcept { return XSIMD_WITH_AVX512VBMI2; }
+        static constexpr bool available() noexcept { return true; }
+        static constexpr char const* name() noexcept { return "avx512vbmi2"; }
+    };
+
+#if XSIMD_WITH_AVX512VBMI2
+
+#if !XSIMD_WITH_AVX512VBMI
+#error "architecture inconsistency: avx512vbmi2 requires avx512vbmi"
+#endif
+
+    namespace types
+    {
+        template <class T>
+        struct get_bool_simd_register<T, avx512vbmi2>
+        {
+            using type = simd_avx512_bool_register<T>;
+        };
+
+        XSIMD_DECLARE_SIMD_REGISTER_ALIAS(avx512vbmi2, avx512vbmi);
+
+    }
+#endif
+}
+#endif