Support SVE runtime dispatcher in xsimd

Change-Id: I2659ca90ae1fedd23ab0fba16882f5f31f36fb0e Signed-off-by: Yuqi Gu <yuqi.gu@arm.com>
xtensor-stack · Mar 22, 2023 · 5c15ff1 · 5c15ff1
1 parent 1a8c4df
commit 5c15ff1
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/include/xsimd/config/xsimd_cpuid.hpp b/include/xsimd/config/xsimd_cpuid.hpp
@@ -52,6 +52,7 @@ namespace xsimd
             unsigned avx512bw : 1;
             unsigned neon : 1;
             unsigned neon64 : 1;
+            unsigned sve : 1;
 
             // version number of the best arch available
             unsigned best;
@@ -75,6 +76,15 @@ namespace xsimd
                 neon64 = 0;
                 best = neon::version() * neon;
 
+#elif defined(__ARM_FEATURE_SVE) && defined(__ARM_FEATURE_SVE_BITS) && __ARM_FEATURE_SVE_BITS > 0
+
+#if defined(__linux__) && (!defined(__ANDROID_API__) || __ANDROID_API__ >= 18)
+                sve = bool(getauxval(AT_HWCAP) & HWCAP_SVE);
+#else
+                sve = 0;
+#endif
+                best = sve::version() * sve;
+
 #elif defined(__x86_64__) || defined(__i386__) || defined(_M_AMD64) || defined(_M_IX86)
                 auto get_cpuid = [](int reg[4], int func_id) noexcept
                 {