Skip to content

Commit 2dd7705

Browse files
[libc] add cpu feature flags for SVE/SVE2/MOPS (#166884)
Add in SVE/SVE2/MOPS features for aarch64 cpus. These features may be interesting for future memory/math routines. SVE/SVE2 are now being accepted in more implementations: ``` ❯ echo | clang-21 -dM -E - -march=native | grep -i ARM_FEAT #define __ARM_FEATURE_ATOMICS 1 #define __ARM_FEATURE_BF16 1 #define __ARM_FEATURE_BF16_SCALAR_ARITHMETIC 1 #define __ARM_FEATURE_BF16_VECTOR_ARITHMETIC 1 #define __ARM_FEATURE_BTI 1 #define __ARM_FEATURE_CLZ 1 #define __ARM_FEATURE_COMPLEX 1 #define __ARM_FEATURE_CRC32 1 #define __ARM_FEATURE_DIRECTED_ROUNDING 1 #define __ARM_FEATURE_DIV 1 #define __ARM_FEATURE_DOTPROD 1 #define __ARM_FEATURE_FMA 1 #define __ARM_FEATURE_FP16_SCALAR_ARITHMETIC 1 #define __ARM_FEATURE_FP16_VECTOR_ARITHMETIC 1 #define __ARM_FEATURE_FRINT 1 #define __ARM_FEATURE_IDIV 1 #define __ARM_FEATURE_JCVT 1 #define __ARM_FEATURE_LDREX 0xF #define __ARM_FEATURE_MATMUL_INT8 1 #define __ARM_FEATURE_NUMERIC_MAXMIN 1 #define __ARM_FEATURE_PAUTH 1 #define __ARM_FEATURE_QRDMX 1 #define __ARM_FEATURE_RCPC 1 #define __ARM_FEATURE_SVE 1 #define __ARM_FEATURE_SVE2 1 #define __ARM_FEATURE_SVE_BF16 1 #define __ARM_FEATURE_SVE_MATMUL_INT8 1 #define __ARM_FEATURE_SVE_VECTOR_OPERATORS 2 #define __ARM_FEATURE_UNALIGNED 1 ``` MOPS is another set of extension for string operations, but may not be generally available for now: ``` ❯ echo | clang-21 -dM -E - -march=armv9.2a+mops | grep -i MOPS #define __ARM_FEATURE_MOPS 1 ```
1 parent c6969e5 commit 2dd7705

File tree

5 files changed

+28
-1
lines changed

5 files changed

+28
-1
lines changed

libc/cmake/modules/LLVMLibCCheckCpuFeatures.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ if(LIBC_TARGET_ARCHITECTURE_IS_X86_64)
99
set(ALL_CPU_FEATURES SSE2 SSE4_2 AVX AVX2 AVX512F AVX512BW FMA)
1010
set(LIBC_COMPILE_OPTIONS_NATIVE -march=native)
1111
elseif(LIBC_TARGET_ARCHITECTURE_IS_AARCH64)
12-
set(ALL_CPU_FEATURES "FullFP16")
12+
set(ALL_CPU_FEATURES FullFP16 MOPS SVE SVE2)
1313
set(LIBC_COMPILE_OPTIONS_NATIVE -mcpu=native)
1414
endif()
1515

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
#include "src/__support/macros/properties/cpu_features.h"
2+
3+
#ifndef LIBC_TARGET_CPU_HAS_MOPS
4+
#error unsupported
5+
#endif
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
#include "src/__support/macros/properties/cpu_features.h"
2+
3+
#ifndef LIBC_TARGET_CPU_HAS_SVE
4+
#error unsupported
5+
#endif
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
#include "src/__support/macros/properties/cpu_features.h"
2+
3+
#ifndef LIBC_TARGET_CPU_HAS_SVE2
4+
#error unsupported
5+
#endif

libc/src/__support/macros/properties/cpu_features.h

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,18 @@
1818
#define LIBC_TARGET_CPU_HAS_FULLFP16
1919
#endif
2020

21+
#if defined(__ARM_FEATURE_SVE)
22+
#define LIBC_TARGET_CPU_HAS_SVE
23+
#endif
24+
25+
#if defined(__ARM_FEATURE_SVE2)
26+
#define LIBC_TARGET_CPU_HAS_SVE2
27+
#endif
28+
29+
#if defined(__ARM_FEATURE_MOPS)
30+
#define LIBC_TARGET_CPU_HAS_MOPS
31+
#endif
32+
2133
#if defined(__SSE2__)
2234
#define LIBC_TARGET_CPU_HAS_SSE2
2335
#define LIBC_TARGET_CPU_HAS_FPU_FLOAT

0 commit comments

Comments
 (0)