1 // NOTE: Assertions have been autogenerated by utils/update_cc_test_checks.py
2 // RUN: %clang_cc1 -triple thumbv8.1m.main-none-none-eabi -target-feature +mve.fp -mfloat-abi hard -fallow-half-arguments-and-returns -O0 -disable-O0-optnone -S -emit-llvm -o - %s | opt -S -mem2reg -sroa | FileCheck %s
3 // RUN: %clang_cc1 -triple thumbv8.1m.main-none-none-eabi -target-feature +mve.fp -mfloat-abi hard -fallow-half-arguments-and-returns -O0 -disable-O0-optnone -DPOLYMORPHIC -S -emit-llvm -o - %s | opt -S -mem2reg -sroa | FileCheck %s
4 
5 #include <arm_mve.h>
6 
7 // CHECK-LABEL: @test_vabavq_s8(
8 // CHECK-NEXT:  entry:
9 // CHECK-NEXT:    [[TMP0:%.*]] = call i32 @llvm.arm.mve.vabav.v16i8(i32 0, i32 [[A:%.*]], <16 x i8> [[B:%.*]], <16 x i8> [[C:%.*]])
10 // CHECK-NEXT:    ret i32 [[TMP0]]
11 //
test_vabavq_s8(uint32_t a,int8x16_t b,int8x16_t c)12 uint32_t test_vabavq_s8(uint32_t a, int8x16_t b, int8x16_t c) {
13 #ifdef POLYMORPHIC
14   return vabavq(a, b, c);
15 #else
16   return vabavq_s8(a, b, c);
17 #endif
18 }
19 
20 // CHECK-LABEL: @test_vabavq_s16(
21 // CHECK-NEXT:  entry:
22 // CHECK-NEXT:    [[TMP0:%.*]] = call i32 @llvm.arm.mve.vabav.v8i16(i32 0, i32 [[A:%.*]], <8 x i16> [[B:%.*]], <8 x i16> [[C:%.*]])
23 // CHECK-NEXT:    ret i32 [[TMP0]]
24 //
test_vabavq_s16(uint32_t a,int16x8_t b,int16x8_t c)25 uint32_t test_vabavq_s16(uint32_t a, int16x8_t b, int16x8_t c) {
26 #ifdef POLYMORPHIC
27   return vabavq(a, b, c);
28 #else
29   return vabavq_s16(a, b, c);
30 #endif
31 }
32 
33 // CHECK-LABEL: @test_vabavq_s32(
34 // CHECK-NEXT:  entry:
35 // CHECK-NEXT:    [[TMP0:%.*]] = call i32 @llvm.arm.mve.vabav.v4i32(i32 0, i32 [[A:%.*]], <4 x i32> [[B:%.*]], <4 x i32> [[C:%.*]])
36 // CHECK-NEXT:    ret i32 [[TMP0]]
37 //
test_vabavq_s32(uint32_t a,int32x4_t b,int32x4_t c)38 uint32_t test_vabavq_s32(uint32_t a, int32x4_t b, int32x4_t c) {
39 #ifdef POLYMORPHIC
40   return vabavq(a, b, c);
41 #else
42   return vabavq_s32(a, b, c);
43 #endif
44 }
45 
46 // CHECK-LABEL: @test_vabavq_u8(
47 // CHECK-NEXT:  entry:
48 // CHECK-NEXT:    [[TMP0:%.*]] = call i32 @llvm.arm.mve.vabav.v16i8(i32 1, i32 [[A:%.*]], <16 x i8> [[B:%.*]], <16 x i8> [[C:%.*]])
49 // CHECK-NEXT:    ret i32 [[TMP0]]
50 //
test_vabavq_u8(uint32_t a,uint8x16_t b,uint8x16_t c)51 uint32_t test_vabavq_u8(uint32_t a, uint8x16_t b, uint8x16_t c) {
52 #ifdef POLYMORPHIC
53   return vabavq(a, b, c);
54 #else
55   return vabavq_u8(a, b, c);
56 #endif
57 }
58 
59 // CHECK-LABEL: @test_vabavq_u16(
60 // CHECK-NEXT:  entry:
61 // CHECK-NEXT:    [[TMP0:%.*]] = call i32 @llvm.arm.mve.vabav.v8i16(i32 1, i32 [[A:%.*]], <8 x i16> [[B:%.*]], <8 x i16> [[C:%.*]])
62 // CHECK-NEXT:    ret i32 [[TMP0]]
63 //
test_vabavq_u16(uint32_t a,uint16x8_t b,uint16x8_t c)64 uint32_t test_vabavq_u16(uint32_t a, uint16x8_t b, uint16x8_t c) {
65 #ifdef POLYMORPHIC
66   return vabavq(a, b, c);
67 #else
68   return vabavq_u16(a, b, c);
69 #endif
70 }
71 
72 // CHECK-LABEL: @test_vabavq_u32(
73 // CHECK-NEXT:  entry:
74 // CHECK-NEXT:    [[TMP0:%.*]] = call i32 @llvm.arm.mve.vabav.v4i32(i32 1, i32 [[A:%.*]], <4 x i32> [[B:%.*]], <4 x i32> [[C:%.*]])
75 // CHECK-NEXT:    ret i32 [[TMP0]]
76 //
test_vabavq_u32(uint32_t a,uint32x4_t b,uint32x4_t c)77 uint32_t test_vabavq_u32(uint32_t a, uint32x4_t b, uint32x4_t c) {
78 #ifdef POLYMORPHIC
79   return vabavq(a, b, c);
80 #else
81   return vabavq_u32(a, b, c);
82 #endif
83 }
84 
85 // CHECK-LABEL: @test_vabavq_p_s8(
86 // CHECK-NEXT:  entry:
87 // CHECK-NEXT:    [[TMP0:%.*]] = zext i16 [[P:%.*]] to i32
88 // CHECK-NEXT:    [[TMP1:%.*]] = call <16 x i1> @llvm.arm.mve.pred.i2v.v16i1(i32 [[TMP0]])
89 // CHECK-NEXT:    [[TMP2:%.*]] = call i32 @llvm.arm.mve.vabav.predicated.v16i8.v16i1(i32 0, i32 [[A:%.*]], <16 x i8> [[B:%.*]], <16 x i8> [[C:%.*]], <16 x i1> [[TMP1]])
90 // CHECK-NEXT:    ret i32 [[TMP2]]
91 //
test_vabavq_p_s8(uint32_t a,int8x16_t b,int8x16_t c,mve_pred16_t p)92 uint32_t test_vabavq_p_s8(uint32_t a, int8x16_t b, int8x16_t c, mve_pred16_t p) {
93 #ifdef POLYMORPHIC
94   return vabavq_p(a, b, c, p);
95 #else
96   return vabavq_p_s8(a, b, c, p);
97 #endif
98 }
99 
100 // CHECK-LABEL: @test_vabavq_p_s16(
101 // CHECK-NEXT:  entry:
102 // CHECK-NEXT:    [[TMP0:%.*]] = zext i16 [[P:%.*]] to i32
103 // CHECK-NEXT:    [[TMP1:%.*]] = call <8 x i1> @llvm.arm.mve.pred.i2v.v8i1(i32 [[TMP0]])
104 // CHECK-NEXT:    [[TMP2:%.*]] = call i32 @llvm.arm.mve.vabav.predicated.v8i16.v8i1(i32 0, i32 [[A:%.*]], <8 x i16> [[B:%.*]], <8 x i16> [[C:%.*]], <8 x i1> [[TMP1]])
105 // CHECK-NEXT:    ret i32 [[TMP2]]
106 //
test_vabavq_p_s16(uint32_t a,int16x8_t b,int16x8_t c,mve_pred16_t p)107 uint32_t test_vabavq_p_s16(uint32_t a, int16x8_t b, int16x8_t c, mve_pred16_t p) {
108 #ifdef POLYMORPHIC
109   return vabavq_p(a, b, c, p);
110 #else
111   return vabavq_p_s16(a, b, c, p);
112 #endif
113 }
114 
115 // CHECK-LABEL: @test_vabavq_p_s32(
116 // CHECK-NEXT:  entry:
117 // CHECK-NEXT:    [[TMP0:%.*]] = zext i16 [[P:%.*]] to i32
118 // CHECK-NEXT:    [[TMP1:%.*]] = call <4 x i1> @llvm.arm.mve.pred.i2v.v4i1(i32 [[TMP0]])
119 // CHECK-NEXT:    [[TMP2:%.*]] = call i32 @llvm.arm.mve.vabav.predicated.v4i32.v4i1(i32 0, i32 [[A:%.*]], <4 x i32> [[B:%.*]], <4 x i32> [[C:%.*]], <4 x i1> [[TMP1]])
120 // CHECK-NEXT:    ret i32 [[TMP2]]
121 //
test_vabavq_p_s32(uint32_t a,int32x4_t b,int32x4_t c,mve_pred16_t p)122 uint32_t test_vabavq_p_s32(uint32_t a, int32x4_t b, int32x4_t c, mve_pred16_t p) {
123 #ifdef POLYMORPHIC
124   return vabavq_p(a, b, c, p);
125 #else
126   return vabavq_p_s32(a, b, c, p);
127 #endif
128 }
129 
130 // CHECK-LABEL: @test_vabavq_p_u8(
131 // CHECK-NEXT:  entry:
132 // CHECK-NEXT:    [[TMP0:%.*]] = zext i16 [[P:%.*]] to i32
133 // CHECK-NEXT:    [[TMP1:%.*]] = call <16 x i1> @llvm.arm.mve.pred.i2v.v16i1(i32 [[TMP0]])
134 // CHECK-NEXT:    [[TMP2:%.*]] = call i32 @llvm.arm.mve.vabav.predicated.v16i8.v16i1(i32 1, i32 [[A:%.*]], <16 x i8> [[B:%.*]], <16 x i8> [[C:%.*]], <16 x i1> [[TMP1]])
135 // CHECK-NEXT:    ret i32 [[TMP2]]
136 //
test_vabavq_p_u8(uint32_t a,uint8x16_t b,uint8x16_t c,mve_pred16_t p)137 uint32_t test_vabavq_p_u8(uint32_t a, uint8x16_t b, uint8x16_t c, mve_pred16_t p) {
138 #ifdef POLYMORPHIC
139   return vabavq_p(a, b, c, p);
140 #else
141   return vabavq_p_u8(a, b, c, p);
142 #endif
143 }
144 
145 // CHECK-LABEL: @test_vabavq_p_u16(
146 // CHECK-NEXT:  entry:
147 // CHECK-NEXT:    [[TMP0:%.*]] = zext i16 [[P:%.*]] to i32
148 // CHECK-NEXT:    [[TMP1:%.*]] = call <8 x i1> @llvm.arm.mve.pred.i2v.v8i1(i32 [[TMP0]])
149 // CHECK-NEXT:    [[TMP2:%.*]] = call i32 @llvm.arm.mve.vabav.predicated.v8i16.v8i1(i32 1, i32 [[A:%.*]], <8 x i16> [[B:%.*]], <8 x i16> [[C:%.*]], <8 x i1> [[TMP1]])
150 // CHECK-NEXT:    ret i32 [[TMP2]]
151 //
test_vabavq_p_u16(uint32_t a,uint16x8_t b,uint16x8_t c,mve_pred16_t p)152 uint32_t test_vabavq_p_u16(uint32_t a, uint16x8_t b, uint16x8_t c, mve_pred16_t p) {
153 #ifdef POLYMORPHIC
154   return vabavq_p(a, b, c, p);
155 #else
156   return vabavq_p_u16(a, b, c, p);
157 #endif
158 }
159 
160 // CHECK-LABEL: @test_vabavq_p_u32(
161 // CHECK-NEXT:  entry:
162 // CHECK-NEXT:    [[TMP0:%.*]] = zext i16 [[P:%.*]] to i32
163 // CHECK-NEXT:    [[TMP1:%.*]] = call <4 x i1> @llvm.arm.mve.pred.i2v.v4i1(i32 [[TMP0]])
164 // CHECK-NEXT:    [[TMP2:%.*]] = call i32 @llvm.arm.mve.vabav.predicated.v4i32.v4i1(i32 1, i32 [[A:%.*]], <4 x i32> [[B:%.*]], <4 x i32> [[C:%.*]], <4 x i1> [[TMP1]])
165 // CHECK-NEXT:    ret i32 [[TMP2]]
166 //
test_vabavq_p_u32(uint32_t a,uint32x4_t b,uint32x4_t c,mve_pred16_t p)167 uint32_t test_vabavq_p_u32(uint32_t a, uint32x4_t b, uint32x4_t c, mve_pred16_t p) {
168 #ifdef POLYMORPHIC
169   return vabavq_p(a, b, c, p);
170 #else
171   return vabavq_p_u32(a, b, c, p);
172 #endif
173 }
174