diff --git a/clang/lib/Driver/ToolChains/Arch/AArch64.cpp b/clang/lib/Driver/ToolChains/Arch/AArch64.cpp --- a/clang/lib/Driver/ToolChains/Arch/AArch64.cpp +++ b/clang/lib/Driver/ToolChains/Arch/AArch64.cpp @@ -365,6 +365,10 @@ } } + auto V8_6Pos = llvm::find(Features, "+v8.6a"); + if (V8_6Pos != std::end(Features)) + V8_6Pos = Features.insert(std::next(V8_6Pos), {"+i8mm", "+bf16"}); + if (Arg *A = Args.getLastArg(options::OPT_mno_unaligned_access, options::OPT_munaligned_access)) if (A->getOption().matches(options::OPT_mno_unaligned_access)) diff --git a/clang/test/Preprocessor/aarch64-target-features.c b/clang/test/Preprocessor/aarch64-target-features.c --- a/clang/test/Preprocessor/aarch64-target-features.c +++ b/clang/test/Preprocessor/aarch64-target-features.c @@ -112,6 +112,24 @@ // CHECK-SVE-F64MM: __ARM_FEATURE_SVE 1 // CHECK-SVE-F64MM: __ARM_FEATURE_SVE_MATMUL_FP64 1 +// RUN: %clang -target aarch64-none-linux-gnu -march=armv8.5-a+sve -x c -E -dM %s -o - | FileCheck --check-prefix=CHECK-SVE-8_5 %s +// CHECK-SVE-8_5-NOT: __ARM_FEATURE_SVE_BF16 1 +// CHECK-SVE-8_5-NOT: __ARM_FEATURE_SVE_MATMUL_FP32 1 +// CHECK-SVE-8_5-NOT: __ARM_FEATURE_SVE_MATMUL_INT8 1 +// CHECK-SVE-8_5: __ARM_FEATURE_SVE 1 + +// RUN: %clang -target aarch64-none-linux-gnu -march=armv8.6-a+sve -x c -E -dM %s -o - | FileCheck --check-prefix=CHECK-SVE-8_6 %s +// CHECK-SVE-8_6: __ARM_FEATURE_SVE 1 +// CHECK-SVE-8_6: __ARM_FEATURE_SVE_BF16 1 +// CHECK-SVE-8_6: __ARM_FEATURE_SVE_MATMUL_FP32 1 +// CHECK-SVE-8_6: __ARM_FEATURE_SVE_MATMUL_INT8 1 + +// RUN: %clang -target aarch64-none-linux-gnu -march=armv8.6-a+sve+noi8mm+nobf16+nof32mm -x c -E -dM %s -o - | FileCheck --check-prefix=CHECK-SVE-8_6-NOFEATURES %s +// CHECK-SVE-8_6-NOFEATURES-NOT: __ARM_FEATURE_SVE_BF16 1 +// CHECK-SVE-8_6-NOFEATURES-NOT: __ARM_FEATURE_SVE_MATMUL_FP32 1 +// CHECK-SVE-8_6-NOFEATURES-NOT: __ARM_FEATURE_SVE_MATMUL_INT8 1 +// CHECK-SVE-8_6-NOFEATURES: __ARM_FEATURE_SVE 1 + // The following tests may need to be revised in the future since // SVE2 is currently still part of Future Architecture Technologies // (https://developer.arm.com/docs/ddi0602/latest)