Add Sema testing for +sve+bf16.

Author: paulwalker-arm

clang/test/Sema/aarch64-sve-intrinsics/acle_sve_bf16.cpp

@@ -0,0 +1,71 @@
+// RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -verify -verify-ignore-unexpected=error,note -emit-llvm -o - %s
+// RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sme -verify -verify-ignore-unexpected=error,note -emit-llvm -o - %s
+// REQUIRES: aarch64-registered-target
+
+#include <arm_sve.h>
+
+#if defined __ARM_FEATURE_SME
+#define MODE_ATTR __arm_streaming
+#else
+#define MODE_ATTR
+#endif
+
+__attribute__((target("bf16")))
+void test_bf16(svbool_t pg, svfloat32_t svf32, svbfloat16_t svbf16, bfloat16_t bf16) MODE_ATTR
+{
+  svbfdot_f32(svf32, svbf16, svbf16);
+  svbfdot_n_f32(svf32, svbf16, bf16);
+  svbfdot_lane_f32(svf32, svbf16, svbf16, 0);
+
+  svbfmlalb_f32(svf32, svbf16, svbf16);
+  svbfmlalb_n_f32(svf32, svbf16, bf16);
+  svbfmlalb_lane_f32(svf32, svbf16, svbf16, 0);
+
+  svbfmlalt_f32(svf32, svbf16, svbf16);
+  svbfmlalt_n_f32(svf32, svbf16, bf16);
+  svbfmlalt_lane_f32(svf32, svbf16, svbf16, 0);
+
+  svcvt_bf16_f32_m(svbf16, pg, svf32);
+  svcvt_bf16_f32_x(pg, svf32);
+  svcvt_bf16_f32_z(pg, svf32);
+
+  svcvtnt_bf16_f32_m(svbf16, pg, svf32);
+  svcvtnt_bf16_f32_x(svbf16, pg, svf32);
+}
+
+void test_no_bf16(svbool_t pg, svfloat32_t svf32, svbfloat16_t svbf16, bfloat16_t bf16) MODE_ATTR
+{
+  // expected-error@+1 {{'svbfdot_f32' needs target feature (sve,bf16)|(sme,bf16)}}
+  svbfdot_f32(svf32, svbf16, svbf16);
+  // expected-error@+1 {{'svbfdot_n_f32' needs target feature (sve,bf16)|(sme,bf16)}}
+  svbfdot_n_f32(svf32, svbf16, bf16);
+  // expected-error@+1 {{'svbfdot_lane_f32' needs target feature (sve,bf16)|(sme,bf16)}}
+  svbfdot_lane_f32(svf32, svbf16, svbf16, 0);
+
+  // expected-error@+1 {{'svbfmlalb_f32' needs target feature (sve,bf16)|(sme,bf16)}}
+  svbfmlalb_f32(svf32, svbf16, svbf16);
+  // expected-error@+1 {{'svbfmlalb_n_f32' needs target feature (sve,bf16)|(sme,bf16)}}
+  svbfmlalb_n_f32(svf32, svbf16, bf16);
+  // expected-error@+1 {{'svbfmlalb_lane_f32' needs target feature (sve,bf16)|(sme,bf16)}}
+  svbfmlalb_lane_f32(svf32, svbf16, svbf16, 0);
+
+  // expected-error@+1 {{'svbfmlalt_f32' needs target feature (sve,bf16)|(sme,bf16)}}
+  svbfmlalt_f32(svf32, svbf16, svbf16);
+  // expected-error@+1 {{'svbfmlalt_n_f32' needs target feature (sve,bf16)|(sme,bf16)}}
+  svbfmlalt_n_f32(svf32, svbf16, bf16);
+  // expected-error@+1 {{'svbfmlalt_lane_f32' needs target feature (sve,bf16)|(sme,bf16)}}
+  svbfmlalt_lane_f32(svf32, svbf16, svbf16, 0);
+
+  // expected-error@+1 {{'svcvt_bf16_f32_m' needs target feature (sve,bf16)|(sme,bf16)}}
+  svcvt_bf16_f32_m(svbf16, pg, svf32);
+  // expected-error@+1 {{'svcvt_bf16_f32_x' needs target feature (sve,bf16)|(sme,bf16)}}
+  svcvt_bf16_f32_x(pg, svf32);
+  // expected-error@+1 {{'svcvt_bf16_f32_z' needs target feature (sve,bf16)|(sme,bf16)}}
+  svcvt_bf16_f32_z(pg, svf32);
+
+  // expected-error@+1 {{'svcvtnt_bf16_f32_m' needs target feature (sve,bf16)|(sme,bf16)}}
+  svcvtnt_bf16_f32_m(svbf16, pg, svf32);
+  // NOTE: svcvtnt_bf16_f32_x is a macro that expands to svcvtnt_bf16_f32_m.
+  // expected-error@+1 {{'svcvtnt_bf16_f32_m' needs target feature (sve,bf16)|(sme,bf16)}}
+  svcvtnt_bf16_f32_x(svbf16, pg, svf32);
+}

clang/test/Sema/aarch64-sve-intrinsics/acle_sve_bf16_non_streaming_only.cpp

@@ -0,0 +1,23 @@
+// RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -verify -verify-ignore-unexpected=error,note -emit-llvm -o - %s
+// REQUIRES: aarch64-registered-target
+
+#include <arm_sve.h>
+
+__attribute__((target("bf16")))
+void test_bf16(svfloat32_t svf32, svbfloat16_t svbf16)
+{
+  svbfmmla_f32(svf32, svbf16, svbf16);
+}
+
+void test_no_bf16(svfloat32_t svf32, svbfloat16_t svbf16)
+{
+  // expected-error@+1 {{'svbfmmla_f32' needs target feature sve,bf16}}
+  svbfmmla_f32(svf32, svbf16, svbf16);
+}
+
+__attribute__((target("sme,bf16")))
+void test_bf16_streaming(svfloat32_t svf32, svbfloat16_t svbf16) __arm_streaming
+{
+  // expected-error@+1 {{builtin can only be called from a non-streaming function}}
+  svbfmmla_f32(svf32, svbf16, svbf16);
+}