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