115 lines · c
1// Test memory tagging extension intrinsics2// RUN: %clang_cc1 -triple aarch64 -target-feature +mte -O3 -emit-llvm -o - %s | FileCheck %s3// RUN: %clang_cc1 -triple aarch64 -DMTE -O3 -emit-llvm -o - %s | FileCheck %s4#include <stddef.h>5#include <arm_acle.h>6 7#ifdef MTE8#define attribute __attribute__((target("mte")))9#else10#define attribute11#endif12 13// CHECK-LABEL: define{{.*}} ptr @create_tag114attribute15int *create_tag1(int *a, unsigned b) {16// CHECK: [[T1:%[0-9]+]] = zext i32 %b to i6417// CHECK: [[T2:%[0-9]+]] = tail call ptr @llvm.aarch64.irg(ptr %a, i64 [[T1]])18 return __arm_mte_create_random_tag(a,b);19}20 21// CHECK-LABEL: define{{.*}} ptr @create_tag222attribute23short *create_tag2(short *a, unsigned b) {24// CHECK: [[T1:%[0-9]+]] = zext i32 %b to i6425// CHECK: [[T2:%[0-9]+]] = tail call ptr @llvm.aarch64.irg(ptr %a, i64 [[T1]])26 return __arm_mte_create_random_tag(a,b);27}28 29// CHECK-LABEL: define{{.*}} ptr @create_tag330attribute31char *create_tag3(char *a, unsigned b) {32// CHECK: [[T1:%[0-9]+]] = zext i32 %b to i6433// CHECK: [[T2:%[0-9]+]] = tail call ptr @llvm.aarch64.irg(ptr %a, i64 [[T1]])34// CHECK: ret ptr [[T2:%[0-9]+]]35 return __arm_mte_create_random_tag(a,b);36}37 38// CHECK-LABEL: define{{.*}} ptr @increment_tag139attribute40char *increment_tag1(char *a) {41// CHECK: call ptr @llvm.aarch64.addg(ptr %a, i64 3)42 return __arm_mte_increment_tag(a,3);43}44 45// CHECK-LABEL: define{{.*}} ptr @increment_tag246attribute47short *increment_tag2(short *a) {48// CHECK: [[T1:%[0-9]+]] = tail call ptr @llvm.aarch64.addg(ptr %a, i64 3)49 return __arm_mte_increment_tag(a,3);50}51 52// CHECK-LABEL: define{{.*}} i32 @exclude_tag53attribute54unsigned exclude_tag(int *a, unsigned m) {55// CHECK: [[T0:%[0-9]+]] = zext i32 %m to i6456// CHECK: [[T2:%[0-9]+]] = tail call i64 @llvm.aarch64.gmi(ptr %a, i64 [[T0]])57// CHECK: trunc i64 [[T2]] to i3258 return __arm_mte_exclude_tag(a, m);59}60 61// CHECK-LABEL: define{{.*}} ptr @get_tag162attribute63int *get_tag1(int *a) {64// CHECK: [[T1:%[0-9]+]] = tail call ptr @llvm.aarch64.ldg(ptr %a, ptr %a)65 return __arm_mte_get_tag(a);66}67 68// CHECK-LABEL: define{{.*}} ptr @get_tag269attribute70short *get_tag2(short *a) {71// CHECK: [[T1:%[0-9]+]] = tail call ptr @llvm.aarch64.ldg(ptr %a, ptr %a)72 return __arm_mte_get_tag(a);73}74 75// CHECK-LABEL: define{{.*}} void @set_tag176attribute77void set_tag1(int *a) {78// CHECK: tail call void @llvm.aarch64.stg(ptr %a, ptr %a)79 __arm_mte_set_tag(a);80}81 82// CHECK-LABEL: define{{.*}} i64 @subtract_pointers83attribute84ptrdiff_t subtract_pointers(int *a, int *b) {85// CHECK: [[T2:%[0-9]+]] = tail call i64 @llvm.aarch64.subp(ptr %a, ptr %b)86// CHECK: ret i64 [[T2]]87 return __arm_mte_ptrdiff(a, b);88}89 90// CHECK-LABEL: define{{.*}} i64 @subtract_pointers_null_191attribute92ptrdiff_t subtract_pointers_null_1(int *a) {93// CHECK: [[T1:%[0-9]+]] = tail call i64 @llvm.aarch64.subp(ptr %a, ptr null)94// CHECK: ret i64 [[T1]]95 return __arm_mte_ptrdiff(a, NULL);96}97 98// CHECK-LABEL: define{{.*}} i64 @subtract_pointers_null_299attribute100ptrdiff_t subtract_pointers_null_2(int *a) {101// CHECK: [[T1:%[0-9]+]] = tail call i64 @llvm.aarch64.subp(ptr null, ptr %a)102// CHECK: ret i64 [[T1]]103 return __arm_mte_ptrdiff(NULL, a);104}105 106// Check arithmetic promotion on return type107// CHECK-LABEL: define{{.*}} i32 @subtract_pointers4108attribute109int subtract_pointers4(void* a, void *b) {110// CHECK: [[T0:%[0-9]+]] = tail call i64 @llvm.aarch64.subp(ptr %a, ptr %b)111// CHECK-NEXT: %cmp = icmp slt i64 [[T0]], 1112// CHECK-NEXT: = zext i1 %cmp to i32113 return __arm_mte_ptrdiff(a,b) <= 0;114}115