135 lines · c
1// NOTE: Assertions have been autogenerated by utils/update_cc_test_checks.py UTC_ARGS: --version 62// RUN: %clang_cc1 -triple aarch64 -target-feature +sve -target-feature +bf16 -mvscale-min=1 -mvscale-max=1 -O1 -emit-llvm -o - %s -fhalf-no-semantic-interposition | FileCheck %s --check-prefix=CHECK-1283// RUN: %clang_cc1 -triple aarch64 -target-feature +sve -target-feature +bf16 -mvscale-min=4 -mvscale-max=4 -O1 -emit-llvm -o - %s -fhalf-no-semantic-interposition | FileCheck %s --check-prefix=CHECK-5124 5// REQUIRES: aarch64-registered-target6 7#include <arm_sve.h>8 9#define N __ARM_FEATURE_SVE_BITS10 11typedef svint64_t fixed_int64_t __attribute__((arm_sve_vector_bits(N)));12typedef svbfloat16_t fixed_bfloat16_t __attribute__((arm_sve_vector_bits(N)));13typedef svbool_t fixed_bool_t __attribute__((arm_sve_vector_bits(N)));14 15fixed_int64_t global_i64;16fixed_bfloat16_t global_bf16;17fixed_bool_t global_bool;18 19//===----------------------------------------------------------------------===//20// WRITES21//===----------------------------------------------------------------------===//22 23// CHECK-128-LABEL: define void @write_global_i64(24// CHECK-128-SAME: <vscale x 2 x i64> [[V:%.*]]) local_unnamed_addr #[[ATTR0:[0-9]+]] {25// CHECK-128-NEXT: [[ENTRY:.*:]]26// CHECK-128-NEXT: [[CAST_FIXED:%.*]] = tail call <2 x i64> @llvm.vector.extract.v2i64.nxv2i64(<vscale x 2 x i64> [[V]], i64 0)27// CHECK-128-NEXT: store <2 x i64> [[CAST_FIXED]], ptr @global_i64, align 16, !tbaa [[CHAR_TBAA6:![0-9]+]]28// CHECK-128-NEXT: ret void29//30// CHECK-512-LABEL: define void @write_global_i64(31// CHECK-512-SAME: <vscale x 2 x i64> [[V:%.*]]) local_unnamed_addr #[[ATTR0:[0-9]+]] {32// CHECK-512-NEXT: [[ENTRY:.*:]]33// CHECK-512-NEXT: [[CAST_FIXED:%.*]] = tail call <8 x i64> @llvm.vector.extract.v8i64.nxv2i64(<vscale x 2 x i64> [[V]], i64 0)34// CHECK-512-NEXT: store <8 x i64> [[CAST_FIXED]], ptr @global_i64, align 16, !tbaa [[CHAR_TBAA6:![0-9]+]]35// CHECK-512-NEXT: ret void36//37void write_global_i64(svint64_t v) { global_i64 = v; }38 39// CHECK-128-LABEL: define void @write_global_bf16(40// CHECK-128-SAME: <vscale x 8 x bfloat> [[V:%.*]]) local_unnamed_addr #[[ATTR0]] {41// CHECK-128-NEXT: [[ENTRY:.*:]]42// CHECK-128-NEXT: [[CAST_FIXED:%.*]] = tail call <8 x bfloat> @llvm.vector.extract.v8bf16.nxv8bf16(<vscale x 8 x bfloat> [[V]], i64 0)43// CHECK-128-NEXT: store <8 x bfloat> [[CAST_FIXED]], ptr @global_bf16, align 16, !tbaa [[CHAR_TBAA6]]44// CHECK-128-NEXT: ret void45//46// CHECK-512-LABEL: define void @write_global_bf16(47// CHECK-512-SAME: <vscale x 8 x bfloat> [[V:%.*]]) local_unnamed_addr #[[ATTR0]] {48// CHECK-512-NEXT: [[ENTRY:.*:]]49// CHECK-512-NEXT: [[CAST_FIXED:%.*]] = tail call <32 x bfloat> @llvm.vector.extract.v32bf16.nxv8bf16(<vscale x 8 x bfloat> [[V]], i64 0)50// CHECK-512-NEXT: store <32 x bfloat> [[CAST_FIXED]], ptr @global_bf16, align 16, !tbaa [[CHAR_TBAA6]]51// CHECK-512-NEXT: ret void52//53void write_global_bf16(svbfloat16_t v) { global_bf16 = v; }54 55// CHECK-128-LABEL: define void @write_global_bool(56// CHECK-128-SAME: <vscale x 16 x i1> [[V:%.*]]) local_unnamed_addr #[[ATTR0]] {57// CHECK-128-NEXT: [[ENTRY:.*:]]58// CHECK-128-NEXT: [[TMP0:%.*]] = bitcast <vscale x 16 x i1> [[V]] to <vscale x 2 x i8>59// CHECK-128-NEXT: [[CAST_FIXED:%.*]] = tail call <2 x i8> @llvm.vector.extract.v2i8.nxv2i8(<vscale x 2 x i8> [[TMP0]], i64 0)60// CHECK-128-NEXT: store <2 x i8> [[CAST_FIXED]], ptr @global_bool, align 2, !tbaa [[CHAR_TBAA6]]61// CHECK-128-NEXT: ret void62//63// CHECK-512-LABEL: define void @write_global_bool(64// CHECK-512-SAME: <vscale x 16 x i1> [[V:%.*]]) local_unnamed_addr #[[ATTR0]] {65// CHECK-512-NEXT: [[ENTRY:.*:]]66// CHECK-512-NEXT: [[TMP0:%.*]] = bitcast <vscale x 16 x i1> [[V]] to <vscale x 2 x i8>67// CHECK-512-NEXT: [[CAST_FIXED:%.*]] = tail call <8 x i8> @llvm.vector.extract.v8i8.nxv2i8(<vscale x 2 x i8> [[TMP0]], i64 0)68// CHECK-512-NEXT: store <8 x i8> [[CAST_FIXED]], ptr @global_bool, align 2, !tbaa [[CHAR_TBAA6]]69// CHECK-512-NEXT: ret void70//71void write_global_bool(svbool_t v) { global_bool = v; }72 73//===----------------------------------------------------------------------===//74// READS75//===----------------------------------------------------------------------===//76 77// CHECK-128-LABEL: define <vscale x 2 x i64> @read_global_i64(78// CHECK-128-SAME: ) local_unnamed_addr #[[ATTR2:[0-9]+]] {79// CHECK-128-NEXT: [[ENTRY:.*:]]80// CHECK-128-NEXT: [[TMP0:%.*]] = load <2 x i64>, ptr @global_i64, align 16, !tbaa [[CHAR_TBAA6]]81// CHECK-128-NEXT: [[CAST_SCALABLE:%.*]] = tail call <vscale x 2 x i64> @llvm.vector.insert.nxv2i64.v2i64(<vscale x 2 x i64> poison, <2 x i64> [[TMP0]], i64 0)82// CHECK-128-NEXT: ret <vscale x 2 x i64> [[CAST_SCALABLE]]83//84// CHECK-512-LABEL: define <vscale x 2 x i64> @read_global_i64(85// CHECK-512-SAME: ) local_unnamed_addr #[[ATTR2:[0-9]+]] {86// CHECK-512-NEXT: [[ENTRY:.*:]]87// CHECK-512-NEXT: [[TMP0:%.*]] = load <8 x i64>, ptr @global_i64, align 16, !tbaa [[CHAR_TBAA6]]88// CHECK-512-NEXT: [[CAST_SCALABLE:%.*]] = tail call <vscale x 2 x i64> @llvm.vector.insert.nxv2i64.v8i64(<vscale x 2 x i64> poison, <8 x i64> [[TMP0]], i64 0)89// CHECK-512-NEXT: ret <vscale x 2 x i64> [[CAST_SCALABLE]]90//91svint64_t read_global_i64() { return global_i64; }92 93// CHECK-128-LABEL: define <vscale x 8 x bfloat> @read_global_bf16(94// CHECK-128-SAME: ) local_unnamed_addr #[[ATTR2]] {95// CHECK-128-NEXT: [[ENTRY:.*:]]96// CHECK-128-NEXT: [[TMP0:%.*]] = load <8 x bfloat>, ptr @global_bf16, align 16, !tbaa [[CHAR_TBAA6]]97// CHECK-128-NEXT: [[CAST_SCALABLE:%.*]] = tail call <vscale x 8 x bfloat> @llvm.vector.insert.nxv8bf16.v8bf16(<vscale x 8 x bfloat> poison, <8 x bfloat> [[TMP0]], i64 0)98// CHECK-128-NEXT: ret <vscale x 8 x bfloat> [[CAST_SCALABLE]]99//100// CHECK-512-LABEL: define <vscale x 8 x bfloat> @read_global_bf16(101// CHECK-512-SAME: ) local_unnamed_addr #[[ATTR2]] {102// CHECK-512-NEXT: [[ENTRY:.*:]]103// CHECK-512-NEXT: [[TMP0:%.*]] = load <32 x bfloat>, ptr @global_bf16, align 16, !tbaa [[CHAR_TBAA6]]104// CHECK-512-NEXT: [[CAST_SCALABLE:%.*]] = tail call <vscale x 8 x bfloat> @llvm.vector.insert.nxv8bf16.v32bf16(<vscale x 8 x bfloat> poison, <32 x bfloat> [[TMP0]], i64 0)105// CHECK-512-NEXT: ret <vscale x 8 x bfloat> [[CAST_SCALABLE]]106//107svbfloat16_t read_global_bf16() { return global_bf16; }108 109// CHECK-128-LABEL: define <vscale x 16 x i1> @read_global_bool(110// CHECK-128-SAME: ) local_unnamed_addr #[[ATTR2]] {111// CHECK-128-NEXT: [[ENTRY:.*:]]112// CHECK-128-NEXT: [[TMP0:%.*]] = load <2 x i8>, ptr @global_bool, align 2, !tbaa [[CHAR_TBAA6]]113// CHECK-128-NEXT: [[CAST_SCALABLE:%.*]] = tail call <vscale x 2 x i8> @llvm.vector.insert.nxv2i8.v2i8(<vscale x 2 x i8> poison, <2 x i8> [[TMP0]], i64 0)114// CHECK-128-NEXT: [[TMP1:%.*]] = bitcast <vscale x 2 x i8> [[CAST_SCALABLE]] to <vscale x 16 x i1>115// CHECK-128-NEXT: ret <vscale x 16 x i1> [[TMP1]]116//117// CHECK-512-LABEL: define <vscale x 16 x i1> @read_global_bool(118// CHECK-512-SAME: ) local_unnamed_addr #[[ATTR2]] {119// CHECK-512-NEXT: [[ENTRY:.*:]]120// CHECK-512-NEXT: [[TMP0:%.*]] = load <8 x i8>, ptr @global_bool, align 2, !tbaa [[CHAR_TBAA6]]121// CHECK-512-NEXT: [[CAST_SCALABLE:%.*]] = tail call <vscale x 2 x i8> @llvm.vector.insert.nxv2i8.v8i8(<vscale x 2 x i8> poison, <8 x i8> [[TMP0]], i64 0)122// CHECK-512-NEXT: [[TMP1:%.*]] = bitcast <vscale x 2 x i8> [[CAST_SCALABLE]] to <vscale x 16 x i1>123// CHECK-512-NEXT: ret <vscale x 16 x i1> [[TMP1]]124//125svbool_t read_global_bool() { return global_bool; }126//.127// CHECK-128: [[META4:![0-9]+]] = !{!"omnipotent char", [[META5:![0-9]+]], i64 0}128// CHECK-128: [[META5]] = !{!"Simple C/C++ TBAA"}129// CHECK-128: [[CHAR_TBAA6]] = !{[[META4]], [[META4]], i64 0}130//.131// CHECK-512: [[META4:![0-9]+]] = !{!"omnipotent char", [[META5:![0-9]+]], i64 0}132// CHECK-512: [[META5]] = !{!"Simple C/C++ TBAA"}133// CHECK-512: [[CHAR_TBAA6]] = !{[[META4]], [[META4]], i64 0}134//.135