1 // REQUIRES: aarch64-registered-target
2 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - %s | FileCheck %s
3 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - -x c++ %s | FileCheck %s
4 // RUN: %clang_cc1 -DSVE_OVERLOADED_FORMS -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - %s | FileCheck %s
5 // RUN: %clang_cc1 -DSVE_OVERLOADED_FORMS -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -emit-llvm -o - -x c++ %s | FileCheck %s
6 // RUN: %clang_cc1 -triple aarch64-none-linux-gnu -target-feature +sve -fallow-half-arguments-and-returns -S -O1 -Werror -Wall -o - %s >/dev/null
7 #include <arm_sve.h>
8 
9 #ifdef SVE_OVERLOADED_FORMS
10 // A simple used,unused... macro, long enough to represent any SVE builtin.
11 #define SVE_ACLE_FUNC(A1,A2_UNUSED,A3,A4_UNUSED) A1##A3
12 #else
13 #define SVE_ACLE_FUNC(A1,A2,A3,A4) A1##A2##A3##A4
14 #endif
15 
test_svlen_s8(svint8_t op)16 uint64_t test_svlen_s8(svint8_t op)
17 {
18   // CHECK-LABEL: test_svlen_s8
19   // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
20   // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 4
21   // CHECK: ret i64 %[[SHL]]
22   return SVE_ACLE_FUNC(svlen,_s8,,)(op);
23 }
24 
test_svlen_s16(svint16_t op)25 uint64_t test_svlen_s16(svint16_t op)
26 {
27   // CHECK-LABEL: test_svlen_s16
28   // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
29   // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 3
30   // CHECK: ret i64 %[[SHL]]
31   return SVE_ACLE_FUNC(svlen,_s16,,)(op);
32 }
33 
test_svlen_s32(svint32_t op)34 uint64_t test_svlen_s32(svint32_t op)
35 {
36   // CHECK-LABEL: test_svlen_s32
37   // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
38   // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 2
39   // CHECK: ret i64 %[[SHL]]
40   return SVE_ACLE_FUNC(svlen,_s32,,)(op);
41 }
42 
test_svlen_s64(svint64_t op)43 uint64_t test_svlen_s64(svint64_t op)
44 {
45   // CHECK-LABEL: test_svlen_s64
46   // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
47   // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 1
48   // CHECK: ret i64 %[[SHL]]
49   return SVE_ACLE_FUNC(svlen,_s64,,)(op);
50 }
51 
test_svlen_u8(svuint8_t op)52 uint64_t test_svlen_u8(svuint8_t op)
53 {
54   // CHECK-LABEL: test_svlen_u8
55   // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
56   // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 4
57   // CHECK: ret i64 %[[SHL]]
58   return SVE_ACLE_FUNC(svlen,_u8,,)(op);
59 }
60 
test_svlen_u16(svuint16_t op)61 uint64_t test_svlen_u16(svuint16_t op)
62 {
63   // CHECK-LABEL: test_svlen_u16
64   // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
65   // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 3
66   // CHECK: ret i64 %[[SHL]]
67   return SVE_ACLE_FUNC(svlen,_u16,,)(op);
68 }
69 
test_svlen_u32(svuint32_t op)70 uint64_t test_svlen_u32(svuint32_t op)
71 {
72   // CHECK-LABEL: test_svlen_u32
73   // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
74   // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 2
75   // CHECK: ret i64 %[[SHL]]
76   return SVE_ACLE_FUNC(svlen,_u32,,)(op);
77 }
78 
test_svlen_u64(svuint64_t op)79 uint64_t test_svlen_u64(svuint64_t op)
80 {
81   // CHECK-LABEL: test_svlen_u64
82   // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
83   // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 1
84   // CHECK: ret i64 %[[SHL]]
85   return SVE_ACLE_FUNC(svlen,_u64,,)(op);
86 }
87 
test_svlen_f16(svfloat16_t op)88 uint64_t test_svlen_f16(svfloat16_t op)
89 {
90   // CHECK-LABEL: test_svlen_f16
91   // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
92   // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 3
93   // CHECK: ret i64 %[[SHL]]
94   return SVE_ACLE_FUNC(svlen,_f16,,)(op);
95 }
96 
test_svlen_f32(svfloat32_t op)97 uint64_t test_svlen_f32(svfloat32_t op)
98 {
99   // CHECK-LABEL: test_svlen_f32
100   // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
101   // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 2
102   // CHECK: ret i64 %[[SHL]]
103   return SVE_ACLE_FUNC(svlen,_f32,,)(op);
104 }
105 
test_svlen_f64(svfloat64_t op)106 uint64_t test_svlen_f64(svfloat64_t op)
107 {
108   // CHECK-LABEL: test_svlen_f64
109   // CHECK: %[[VSCALE:.*]] = call i64 @llvm.vscale.i64()
110   // CHECK: %[[SHL:.*]] = shl i64 %[[VSCALE]], 1
111   // CHECK: ret i64 %[[SHL]]
112   return SVE_ACLE_FUNC(svlen,_f64,,)(op);
113 }
114