rustc_target: Add various aarch64 features

Add various aarch64 features already supported by LLVM and Linux.

The features are marked as unstable using a newly added symbol, i.e.
aarch64_unstable_target_feature.

Additionally include some comment fixes to ensure consistency of
feature names with the Arm ARM and support for architecture version
target features up to v9.5a.

This commit adds compiler support for the following features:

- FEAT_CSSC
- FEAT_ECV
- FEAT_FAMINMAX
- FEAT_FLAGM2
- FEAT_FP8
- FEAT_FP8DOT2
- FEAT_FP8DOT4
- FEAT_FP8FMA
- FEAT_FPMR
- FEAT_HBC
- FEAT_LSE128
- FEAT_LSE2
- FEAT_LUT
- FEAT_MOPS
- FEAT_LRCPC3
- FEAT_SVE_B16B16
- FEAT_SVE2p1
- FEAT_WFxT

Author: mrkajetanp

compiler/rustc_codegen_llvm/src/llvm_util.rs

@@ -233,6 +233,8 @@ pub fn to_llvm_features<'a>(sess: &Session, s: &'a str) -> LLVMFeature<'a> {
         ("aarch64", "pmuv3") => LLVMFeature::new("perfmon"),
         ("aarch64", "paca") => LLVMFeature::new("pauth"),
         ("aarch64", "pacg") => LLVMFeature::new("pauth"),
+        ("aarch64", "sve-b16b16") => LLVMFeature::new("b16b16"),
+        ("aarch64", "flagm2") => LLVMFeature::new("altnzcv"),
         // Rust ties fp and neon together.
         ("aarch64", "neon") => {
             LLVMFeature::with_dependency("neon", TargetFeatureFoldStrength::Both("fp-armv8"))
@@ -260,6 +262,9 @@ pub fn to_llvm_features<'a>(sess: &Session, s: &'a str) -> LLVMFeature<'a> {
         ("aarch64", "sve2") => {
             LLVMFeature::with_dependency("sve2", TargetFeatureFoldStrength::EnableOnly("neon"))
         }
+        ("aarch64", "sve2p1") => {
+            LLVMFeature::with_dependency("sve2p1", TargetFeatureFoldStrength::EnableOnly("neon"))
+        }
         ("aarch64", "sve2-aes") => {
             LLVMFeature::with_dependency("sve2-aes", TargetFeatureFoldStrength::EnableOnly("neon"))
         }

compiler/rustc_feature/src/unstable.rs

@@ -300,6 +300,7 @@ declare_features! (
     // FIXME: Document these and merge with the list below.
 
     // Unstable `#[target_feature]` directives.
+    (unstable, aarch64_unstable_target_feature, "CURRENT_RUSTC_VERSION", Some(44839)),
     (unstable, aarch64_ver_target_feature, "1.27.0", Some(44839)),
     (unstable, arm_target_feature, "1.27.0", Some(44839)),
     (unstable, avx512_target_feature, "1.27.0", Some(44839)),

compiler/rustc_span/src/symbol.rs

@@ -355,6 +355,7 @@ symbols! {
         _task_context,
         a32,
         aarch64_target_feature,
+        aarch64_unstable_target_feature,
         aarch64_ver_target_feature,
         abi,
         abi_amdgpu_kernel,

compiler/rustc_target/src/target_features.rs

@@ -96,6 +96,8 @@ const AARCH64_ALLOWED_FEATURES: &[(&str, Stability)] = &[
     ("bti", Stable),
     // FEAT_CRC
     ("crc", Stable),
+    // FEAT_CSSC
+    ("cssc", Unstable(sym::aarch64_unstable_target_feature)),
     // FEAT_DIT
     ("dit", Stable),
     // FEAT_DotProd
@@ -104,20 +106,38 @@ const AARCH64_ALLOWED_FEATURES: &[(&str, Stability)] = &[
     ("dpb", Stable),
     // FEAT_DPB2
     ("dpb2", Stable),
+    // FEAT_ECV
+    ("ecv", Unstable(sym::aarch64_unstable_target_feature)),
     // FEAT_F32MM
     ("f32mm", Stable),
     // FEAT_F64MM
     ("f64mm", Stable),
+    // FEAT_FAMINMAX
+    ("faminmax", Unstable(sym::aarch64_unstable_target_feature)),
     // FEAT_FCMA
     ("fcma", Stable),
     // FEAT_FHM
     ("fhm", Stable),
     // FEAT_FLAGM
     ("flagm", Stable),
+    // FEAT_FLAGM2
+    ("flagm2", Unstable(sym::aarch64_unstable_target_feature)),
     // FEAT_FP16
     ("fp16", Stable),
+    // FEAT_FP8
+    ("fp8", Unstable(sym::aarch64_unstable_target_feature)),
+    // FEAT_FP8DOT2
+    ("fp8dot2", Unstable(sym::aarch64_unstable_target_feature)),
+    // FEAT_FP8DOT4
+    ("fp8dot4", Unstable(sym::aarch64_unstable_target_feature)),
+    // FEAT_FP8FMA
+    ("fp8fma", Unstable(sym::aarch64_unstable_target_feature)),
+    // FEAT_FPMR
+    ("fpmr", Unstable(sym::aarch64_unstable_target_feature)),
     // FEAT_FRINTTS
     ("frintts", Stable),
+    // FEAT_HBC
+    ("hbc", Unstable(sym::aarch64_unstable_target_feature)),
     // FEAT_I8MM
     ("i8mm", Stable),
     // FEAT_JSCVT
@@ -126,6 +146,14 @@ const AARCH64_ALLOWED_FEATURES: &[(&str, Stability)] = &[
     ("lor", Stable),
     // FEAT_LSE
     ("lse", Stable),
+    // FEAT_LSE128
+    ("lse128", Unstable(sym::aarch64_unstable_target_feature)),
+    // FEAT_LSE2
+    ("lse2", Unstable(sym::aarch64_unstable_target_feature)),
+    // FEAT_LUT
+    ("lut", Unstable(sym::aarch64_unstable_target_feature)),
+    // FEAT_MOPS
+    ("mops", Unstable(sym::aarch64_unstable_target_feature)),
     // FEAT_MTE & FEAT_MTE2
     ("mte", Stable),
     // FEAT_AdvSimd & FEAT_FP
@@ -138,14 +166,16 @@ const AARCH64_ALLOWED_FEATURES: &[(&str, Stability)] = &[
     ("pan", Stable),
     // FEAT_PMUv3
     ("pmuv3", Stable),
-    // FEAT_RAND
+    // FEAT_RNG
     ("rand", Stable),
     // FEAT_RAS & FEAT_RASv1p1
     ("ras", Stable),
-    // FEAT_RCPC
+    // FEAT_LRCPC
     ("rcpc", Stable),
-    // FEAT_RCPC2
+    // FEAT_LRCPC2
     ("rcpc2", Stable),
+    // FEAT_LRCPC3
+    ("rcpc3", Unstable(sym::aarch64_unstable_target_feature)),
     // FEAT_RDM
     ("rdm", Stable),
     // FEAT_SB
@@ -162,16 +192,20 @@ const AARCH64_ALLOWED_FEATURES: &[(&str, Stability)] = &[
     ("ssbs", Stable),
     // FEAT_SVE
     ("sve", Stable),
+    // FEAT_SVE_B16B16 (SVE or SME Instructions)
+    ("sve-b16b16", Unstable(sym::aarch64_unstable_target_feature)),
     // FEAT_SVE2
     ("sve2", Stable),
-    // FEAT_SVE2_AES
+    // FEAT_SVE_AES & FEAT_SVE_PMULL128
     ("sve2-aes", Stable),
     // FEAT_SVE2_BitPerm
     ("sve2-bitperm", Stable),
     // FEAT_SVE2_SHA3
     ("sve2-sha3", Stable),
     // FEAT_SVE2_SM4
     ("sve2-sm4", Stable),
+    // FEAT_SVE2p1
+    ("sve2p1", Unstable(sym::aarch64_unstable_target_feature)),
     // FEAT_TME
     ("tme", Stable),
     ("v8.1a", Unstable(sym::aarch64_ver_target_feature)),
@@ -181,8 +215,18 @@ const AARCH64_ALLOWED_FEATURES: &[(&str, Stability)] = &[
     ("v8.5a", Unstable(sym::aarch64_ver_target_feature)),
     ("v8.6a", Unstable(sym::aarch64_ver_target_feature)),
     ("v8.7a", Unstable(sym::aarch64_ver_target_feature)),
+    ("v8.8a", Unstable(sym::aarch64_ver_target_feature)),
+    ("v8.9a", Unstable(sym::aarch64_ver_target_feature)),
+    ("v9.1a", Unstable(sym::aarch64_ver_target_feature)),
+    ("v9.2a", Unstable(sym::aarch64_ver_target_feature)),
+    ("v9.3a", Unstable(sym::aarch64_ver_target_feature)),
+    ("v9.4a", Unstable(sym::aarch64_ver_target_feature)),
+    ("v9.5a", Unstable(sym::aarch64_ver_target_feature)),
+    ("v9a", Unstable(sym::aarch64_ver_target_feature)),
     // FEAT_VHE
     ("vh", Stable),
+    // FEAT_WFxT
+    ("wfxt", Unstable(sym::aarch64_unstable_target_feature)),
     // tidy-alphabetical-end
 ];
 

library/std/tests/run-time-detect.rs

@@ -4,6 +4,10 @@
     all(target_arch = "arm", any(target_os = "linux", target_os = "android")),
     feature(stdarch_arm_feature_detection)
 )]
+#![cfg_attr(
+    all(target_arch = "aarch64", any(target_os = "linux", target_os = "android")),
+    feature(stdarch_aarch64_feature_detection)
+)]
 #![cfg_attr(
     all(target_arch = "powerpc", target_os = "linux"),
     feature(stdarch_powerpc_feature_detection)
@@ -36,42 +40,59 @@ fn aarch64_linux() {
     println!("bf16: {}", is_aarch64_feature_detected!("bf16"));
     println!("bti: {}", is_aarch64_feature_detected!("bti"));
     println!("crc: {}", is_aarch64_feature_detected!("crc"));
+    println!("cssc: {}", is_aarch64_feature_detected!("cssc"));
     println!("dit: {}", is_aarch64_feature_detected!("dit"));
     println!("dotprod: {}", is_aarch64_feature_detected!("dotprod"));
     println!("dpb2: {}", is_aarch64_feature_detected!("dpb2"));
     println!("dpb: {}", is_aarch64_feature_detected!("dpb"));
+    println!("ecv: {}", is_aarch64_feature_detected!("ecv"));
     println!("f32mm: {}", is_aarch64_feature_detected!("f32mm"));
     println!("f64mm: {}", is_aarch64_feature_detected!("f64mm"));
+    println!("faminmax: {}", is_aarch64_feature_detected!("faminmax"));
     println!("fcma: {}", is_aarch64_feature_detected!("fcma"));
     println!("fhm: {}", is_aarch64_feature_detected!("fhm"));
+    println!("flagm2: {}", is_aarch64_feature_detected!("flagm2"));
     println!("flagm: {}", is_aarch64_feature_detected!("flagm"));
     println!("fp16: {}", is_aarch64_feature_detected!("fp16"));
+    println!("fp8: {}", is_aarch64_feature_detected!("fp8"));
+    println!("fp8dot2: {}", is_aarch64_feature_detected!("fp8dot2"));
+    println!("fp8dot4: {}", is_aarch64_feature_detected!("fp8dot4"));
+    println!("fp8fma: {}", is_aarch64_feature_detected!("fp8fma"));
+    println!("fpmr: {}", is_aarch64_feature_detected!("fpmr"));
     println!("frintts: {}", is_aarch64_feature_detected!("frintts"));
+    println!("hbc: {}", is_aarch64_feature_detected!("hbc"));
     println!("i8mm: {}", is_aarch64_feature_detected!("i8mm"));
     println!("jsconv: {}", is_aarch64_feature_detected!("jsconv"));
+    println!("lse128: {}", is_aarch64_feature_detected!("lse128"));
     println!("lse2: {}", is_aarch64_feature_detected!("lse2"));
     println!("lse: {}", is_aarch64_feature_detected!("lse"));
+    println!("lut: {}", is_aarch64_feature_detected!("lut"));
+    println!("mops: {}", is_aarch64_feature_detected!("mops"));
     println!("mte: {}", is_aarch64_feature_detected!("mte"));
     println!("neon: {}", is_aarch64_feature_detected!("neon"));
     println!("paca: {}", is_aarch64_feature_detected!("paca"));
     println!("pacg: {}", is_aarch64_feature_detected!("pacg"));
     println!("pmull: {}", is_aarch64_feature_detected!("pmull"));
     println!("rand: {}", is_aarch64_feature_detected!("rand"));
     println!("rcpc2: {}", is_aarch64_feature_detected!("rcpc2"));
+    println!("rcpc3: {}", is_aarch64_feature_detected!("rcpc3"));
     println!("rcpc: {}", is_aarch64_feature_detected!("rcpc"));
     println!("rdm: {}", is_aarch64_feature_detected!("rdm"));
     println!("sb: {}", is_aarch64_feature_detected!("sb"));
     println!("sha2: {}", is_aarch64_feature_detected!("sha2"));
     println!("sha3: {}", is_aarch64_feature_detected!("sha3"));
     println!("sm4: {}", is_aarch64_feature_detected!("sm4"));
     println!("ssbs: {}", is_aarch64_feature_detected!("ssbs"));
+    println!("sve-b16b16: {}", is_aarch64_feature_detected!("sve-b16b16"));
     println!("sve2-aes: {}", is_aarch64_feature_detected!("sve2-aes"));
     println!("sve2-bitperm: {}", is_aarch64_feature_detected!("sve2-bitperm"));
     println!("sve2-sha3: {}", is_aarch64_feature_detected!("sve2-sha3"));
     println!("sve2-sm4: {}", is_aarch64_feature_detected!("sve2-sm4"));
     println!("sve2: {}", is_aarch64_feature_detected!("sve2"));
+    println!("sve2p1: {}", is_aarch64_feature_detected!("sve2p1"));
     println!("sve: {}", is_aarch64_feature_detected!("sve"));
     println!("tme: {}", is_aarch64_feature_detected!("tme"));
+    println!("wfxt: {}", is_aarch64_feature_detected!("wfxt"));
     // tidy-alphabetical-end
 }
 

tests/ui/target-feature/gate.rs

@@ -17,6 +17,7 @@
 // gate-test-ermsb_target_feature
 // gate-test-bpf_target_feature
 // gate-test-aarch64_ver_target_feature
+// gate-test-aarch64_unstable_target_feature
 // gate-test-csky_target_feature
 // gate-test-loongarch_target_feature
 // gate-test-lahfsahf_target_feature