1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2 ; RUN: llc < %s -mtriple=x86_64-apple-darwin -mcpu=cannonlake | FileCheck %s
4 ; These test cases demonstrate cases where vpermt2/vpermi2 could benefit from being commuted.
6 declare <16 x i32> @llvm.x86.avx512.mask.vpermi2var.d.512(<16 x i32>, <16 x i32>, <16 x i32>, i16)
8 define <16 x i32>@test_int_x86_avx512_mask_vpermi2var_d_512(<16 x i32> %x0, <16 x i32> %x1, ptr %x2p) {
9 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_d_512:
11 ; CHECK-NEXT: vpermt2d (%rdi), %zmm1, %zmm0
13 %x2 = load <16 x i32>, ptr %x2p
14 %res = call <16 x i32> @llvm.x86.avx512.mask.vpermi2var.d.512(<16 x i32> %x0, <16 x i32> %x1, <16 x i32> %x2, i16 -1)
18 declare <8 x double> @llvm.x86.avx512.mask.vpermi2var.pd.512(<8 x double>, <8 x i64>, <8 x double>, i8)
20 define <8 x double>@test_int_x86_avx512_mask_vpermi2var_pd_512(<8 x double> %x0, <8 x i64> %x1, <8 x double> %x2) {
21 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_pd_512:
23 ; CHECK-NEXT: vpermt2pd %zmm2, %zmm1, %zmm0
25 %res = call <8 x double> @llvm.x86.avx512.mask.vpermi2var.pd.512(<8 x double> %x0, <8 x i64> %x1, <8 x double> %x2, i8 -1)
29 declare <16 x float> @llvm.x86.avx512.mask.vpermi2var.ps.512(<16 x float>, <16 x i32>, <16 x float>, i16)
31 define <16 x float>@test_int_x86_avx512_mask_vpermi2var_ps_512(<16 x float> %x0, <16 x i32> %x1, <16 x float> %x2) {
32 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_ps_512:
34 ; CHECK-NEXT: vpermt2ps %zmm2, %zmm1, %zmm0
36 %res = call <16 x float> @llvm.x86.avx512.mask.vpermi2var.ps.512(<16 x float> %x0, <16 x i32> %x1, <16 x float> %x2, i16 -1)
40 declare <8 x i64> @llvm.x86.avx512.mask.vpermi2var.q.512(<8 x i64>, <8 x i64>, <8 x i64>, i8)
42 define <8 x i64>@test_int_x86_avx512_mask_vpermi2var_q_512(<8 x i64> %x0, <8 x i64> %x1, <8 x i64> %x2) {
43 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_q_512:
45 ; CHECK-NEXT: vpermt2q %zmm2, %zmm1, %zmm0
47 %res = call <8 x i64> @llvm.x86.avx512.mask.vpermi2var.q.512(<8 x i64> %x0, <8 x i64> %x1, <8 x i64> %x2, i8 -1)
51 declare <16 x i32> @llvm.x86.avx512.maskz.vpermt2var.d.512(<16 x i32>, <16 x i32>, <16 x i32>, i16)
53 define <16 x i32>@test_int_x86_avx512_maskz_vpermt2var_d_512(<16 x i32> %x0, <16 x i32> %x1, ptr %x2p, i16 %x3) {
54 ; CHECK-LABEL: test_int_x86_avx512_maskz_vpermt2var_d_512:
56 ; CHECK-NEXT: kmovd %esi, %k1
57 ; CHECK-NEXT: vpermi2d (%rdi), %zmm1, %zmm0 {%k1} {z}
59 %x2 = load <16 x i32>, ptr %x2p
60 %res = call <16 x i32> @llvm.x86.avx512.maskz.vpermt2var.d.512(<16 x i32> %x0, <16 x i32> %x1, <16 x i32> %x2, i16 %x3)
64 declare <8 x double> @llvm.x86.avx512.maskz.vpermt2var.pd.512(<8 x i64>, <8 x double>, <8 x double>, i8)
66 define <8 x double>@test_int_x86_avx512_maskz_vpermt2var_pd_512(<8 x i64> %x0, <8 x double> %x1, ptr %x2ptr, i8 %x3) {
67 ; CHECK-LABEL: test_int_x86_avx512_maskz_vpermt2var_pd_512:
69 ; CHECK-NEXT: kmovd %esi, %k1
70 ; CHECK-NEXT: vpermi2pd (%rdi){1to8}, %zmm1, %zmm0 {%k1} {z}
72 %x2s = load double, ptr %x2ptr
73 %x2ins = insertelement <8 x double> undef, double %x2s, i32 0
74 %x2 = shufflevector <8 x double> %x2ins, <8 x double> undef, <8 x i32> zeroinitializer
75 %res = call <8 x double> @llvm.x86.avx512.maskz.vpermt2var.pd.512(<8 x i64> %x0, <8 x double> %x1, <8 x double> %x2, i8 %x3)
79 declare <16 x float> @llvm.x86.avx512.maskz.vpermt2var.ps.512(<16 x i32>, <16 x float>, <16 x float>, i16)
81 define <16 x float>@test_int_x86_avx512_maskz_vpermt2var_ps_512(<16 x i32> %x0, <16 x float> %x1, <16 x float> %x2, i16 %x3) {
82 ; CHECK-LABEL: test_int_x86_avx512_maskz_vpermt2var_ps_512:
84 ; CHECK-NEXT: kmovd %edi, %k1
85 ; CHECK-NEXT: vpermi2ps %zmm2, %zmm1, %zmm0 {%k1} {z}
87 %res = call <16 x float> @llvm.x86.avx512.maskz.vpermt2var.ps.512(<16 x i32> %x0, <16 x float> %x1, <16 x float> %x2, i16 %x3)
92 declare <8 x i64> @llvm.x86.avx512.maskz.vpermt2var.q.512(<8 x i64>, <8 x i64>, <8 x i64>, i8)
94 define <8 x i64>@test_int_x86_avx512_maskz_vpermt2var_q_512(<8 x i64> %x0, <8 x i64> %x1, <8 x i64> %x2, i8 %x3) {
95 ; CHECK-LABEL: test_int_x86_avx512_maskz_vpermt2var_q_512:
97 ; CHECK-NEXT: kmovd %edi, %k1
98 ; CHECK-NEXT: vpermi2q %zmm2, %zmm1, %zmm0 {%k1} {z}
100 %res = call <8 x i64> @llvm.x86.avx512.maskz.vpermt2var.q.512(<8 x i64> %x0, <8 x i64> %x1, <8 x i64> %x2, i8 %x3)
104 declare <16 x i32> @llvm.x86.avx512.mask.vpermt2var.d.512(<16 x i32>, <16 x i32>, <16 x i32>, i16)
106 define <16 x i32>@test_int_x86_avx512_mask_vpermt2var_d_512(<16 x i32> %x0, <16 x i32> %x1, <16 x i32> %x2) {
107 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermt2var_d_512:
109 ; CHECK-NEXT: vpermi2d %zmm2, %zmm1, %zmm0
111 %res = call <16 x i32> @llvm.x86.avx512.mask.vpermt2var.d.512(<16 x i32> %x0, <16 x i32> %x1, <16 x i32> %x2, i16 -1)
115 declare <4 x i32> @llvm.x86.avx512.mask.vpermt2var.d.128(<4 x i32>, <4 x i32>, <4 x i32>, i8)
117 define <4 x i32>@test_int_x86_avx512_mask_vpermt2var_d_128(<4 x i32> %x0, <4 x i32> %x1, <4 x i32> %x2) {
118 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermt2var_d_128:
120 ; CHECK-NEXT: vpermi2d %xmm2, %xmm1, %xmm0
122 %res = call <4 x i32> @llvm.x86.avx512.mask.vpermt2var.d.128(<4 x i32> %x0, <4 x i32> %x1, <4 x i32> %x2, i8 -1)
126 declare <4 x i32> @llvm.x86.avx512.maskz.vpermt2var.d.128(<4 x i32>, <4 x i32>, <4 x i32>, i8)
128 define <4 x i32>@test_int_x86_avx512_maskz_vpermt2var_d_128(<4 x i32> %x0, <4 x i32> %x1, <4 x i32> %x2, i8 %x3) {
129 ; CHECK-LABEL: test_int_x86_avx512_maskz_vpermt2var_d_128:
131 ; CHECK-NEXT: kmovd %edi, %k1
132 ; CHECK-NEXT: vpermi2d %xmm2, %xmm1, %xmm0 {%k1} {z}
134 %res = call <4 x i32> @llvm.x86.avx512.maskz.vpermt2var.d.128(<4 x i32> %x0, <4 x i32> %x1, <4 x i32> %x2, i8 %x3)
138 define <4 x i32>@test_int_x86_avx512_maskz_vpermt2var_d_128_broadcast(<4 x i32> %x0, <4 x i32> %x1, ptr %x2ptr, i8 %x3) {
139 ; CHECK-LABEL: test_int_x86_avx512_maskz_vpermt2var_d_128_broadcast:
141 ; CHECK-NEXT: kmovd %esi, %k1
142 ; CHECK-NEXT: vpermi2d (%rdi){1to4}, %xmm1, %xmm0 {%k1} {z}
144 %x2s = load i32, ptr %x2ptr
145 %x2ins = insertelement <4 x i32> undef, i32 %x2s, i32 0
146 %x2 = shufflevector <4 x i32> %x2ins, <4 x i32> undef, <4 x i32> zeroinitializer
147 %res = call <4 x i32> @llvm.x86.avx512.maskz.vpermt2var.d.128(<4 x i32> %x0, <4 x i32> %x1, <4 x i32> %x2, i8 %x3)
151 declare <8 x i32> @llvm.x86.avx512.mask.vpermt2var.d.256(<8 x i32>, <8 x i32>, <8 x i32>, i8)
153 define <8 x i32>@test_int_x86_avx512_mask_vpermt2var_d_256(<8 x i32> %x0, <8 x i32> %x1, <8 x i32> %x2) {
154 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermt2var_d_256:
156 ; CHECK-NEXT: vpermi2d %ymm2, %ymm1, %ymm0
158 %res = call <8 x i32> @llvm.x86.avx512.mask.vpermt2var.d.256(<8 x i32> %x0, <8 x i32> %x1, <8 x i32> %x2, i8 -1)
162 declare <8 x i32> @llvm.x86.avx512.maskz.vpermt2var.d.256(<8 x i32>, <8 x i32>, <8 x i32>, i8)
164 define <8 x i32>@test_int_x86_avx512_maskz_vpermt2var_d_256(<8 x i32> %x0, <8 x i32> %x1, <8 x i32> %x2, i8 %x3) {
165 ; CHECK-LABEL: test_int_x86_avx512_maskz_vpermt2var_d_256:
167 ; CHECK-NEXT: kmovd %edi, %k1
168 ; CHECK-NEXT: vpermi2d %ymm2, %ymm1, %ymm0 {%k1} {z}
170 %res = call <8 x i32> @llvm.x86.avx512.maskz.vpermt2var.d.256(<8 x i32> %x0, <8 x i32> %x1, <8 x i32> %x2, i8 %x3)
174 declare <2 x double> @llvm.x86.avx512.mask.vpermi2var.pd.128(<2 x double>, <2 x i64>, <2 x double>, i8)
176 define <2 x double>@test_int_x86_avx512_mask_vpermi2var_pd_128(<2 x double> %x0, <2 x i64> %x1, <2 x double> %x2) {
177 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_pd_128:
179 ; CHECK-NEXT: vpermt2pd %xmm2, %xmm1, %xmm0
181 %res = call <2 x double> @llvm.x86.avx512.mask.vpermi2var.pd.128(<2 x double> %x0, <2 x i64> %x1, <2 x double> %x2, i8 -1)
182 ret <2 x double> %res
185 declare <4 x double> @llvm.x86.avx512.mask.vpermi2var.pd.256(<4 x double>, <4 x i64>, <4 x double>, i8)
187 define <4 x double>@test_int_x86_avx512_mask_vpermi2var_pd_256(<4 x double> %x0, <4 x i64> %x1, <4 x double> %x2) {
188 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_pd_256:
190 ; CHECK-NEXT: vpermt2pd %ymm2, %ymm1, %ymm0
192 %res = call <4 x double> @llvm.x86.avx512.mask.vpermi2var.pd.256(<4 x double> %x0, <4 x i64> %x1, <4 x double> %x2, i8 -1)
193 ret <4 x double> %res
196 declare <4 x float> @llvm.x86.avx512.mask.vpermi2var.ps.128(<4 x float>, <4 x i32>, <4 x float>, i8)
198 define <4 x float>@test_int_x86_avx512_mask_vpermi2var_ps_128(<4 x float> %x0, <4 x i32> %x1, <4 x float> %x2) {
199 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_ps_128:
201 ; CHECK-NEXT: vpermt2ps %xmm2, %xmm1, %xmm0
203 %res = call <4 x float> @llvm.x86.avx512.mask.vpermi2var.ps.128(<4 x float> %x0, <4 x i32> %x1, <4 x float> %x2, i8 -1)
207 declare <8 x float> @llvm.x86.avx512.mask.vpermi2var.ps.256(<8 x float>, <8 x i32>, <8 x float>, i8)
209 define <8 x float>@test_int_x86_avx512_mask_vpermi2var_ps_256(<8 x float> %x0, <8 x i32> %x1, <8 x float> %x2) {
210 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_ps_256:
212 ; CHECK-NEXT: vpermt2ps %ymm2, %ymm1, %ymm0
214 %res = call <8 x float> @llvm.x86.avx512.mask.vpermi2var.ps.256(<8 x float> %x0, <8 x i32> %x1, <8 x float> %x2, i8 -1)
218 define <8 x float>@test_int_x86_avx512_mask_vpermi2var_ps_256_load(<8 x float> %x0, <8 x i32> %x1, ptr %x2p) {
219 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_ps_256_load:
221 ; CHECK-NEXT: vpermt2ps (%rdi), %ymm1, %ymm0
223 %x2 = load <8 x float>, ptr %x2p
224 %res = call <8 x float> @llvm.x86.avx512.mask.vpermi2var.ps.256(<8 x float> %x0, <8 x i32> %x1, <8 x float> %x2, i8 -1)
228 define <8 x float>@test_int_x86_avx512_mask_vpermi2var_ps_256_broadcast(<8 x float> %x0, <8 x i32> %x1, ptr %x2ptr) {
229 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_ps_256_broadcast:
231 ; CHECK-NEXT: vpermt2ps (%rdi){1to8}, %ymm1, %ymm0
233 %x2s = load float, ptr %x2ptr
234 %x2ins = insertelement <8 x float> undef, float %x2s, i32 0
235 %x2 = shufflevector <8 x float> %x2ins, <8 x float> undef, <8 x i32> zeroinitializer
236 %res = call <8 x float> @llvm.x86.avx512.mask.vpermi2var.ps.256(<8 x float> %x0, <8 x i32> %x1, <8 x float> %x2, i8 -1)
240 declare <16 x i8> @llvm.x86.avx512.mask.vpermi2var.qi.128(<16 x i8>, <16 x i8>, <16 x i8>, i16)
242 define <16 x i8>@test_int_x86_avx512_mask_vpermi2var_qi_128(<16 x i8> %x0, <16 x i8> %x1, <16 x i8> %x2) {
243 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_qi_128:
245 ; CHECK-NEXT: vpermt2b %xmm2, %xmm1, %xmm0
247 %res = call <16 x i8> @llvm.x86.avx512.mask.vpermi2var.qi.128(<16 x i8> %x0, <16 x i8> %x1, <16 x i8> %x2, i16 -1)
251 declare <32 x i8> @llvm.x86.avx512.mask.vpermi2var.qi.256(<32 x i8>, <32 x i8>, <32 x i8>, i32)
253 define <32 x i8>@test_int_x86_avx512_mask_vpermi2var_qi_256(<32 x i8> %x0, <32 x i8> %x1, <32 x i8> %x2) {
254 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermi2var_qi_256:
256 ; CHECK-NEXT: vpermt2b %ymm2, %ymm1, %ymm0
258 %res = call <32 x i8> @llvm.x86.avx512.mask.vpermi2var.qi.256(<32 x i8> %x0, <32 x i8> %x1, <32 x i8> %x2, i32 -1)
262 declare <16 x i8> @llvm.x86.avx512.mask.vpermt2var.qi.128(<16 x i8>, <16 x i8>, <16 x i8>, i16)
264 define <16 x i8>@test_int_x86_avx512_mask_vpermt2var_qi_128(<16 x i8> %x0, <16 x i8> %x1, <16 x i8> %x2) {
265 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermt2var_qi_128:
267 ; CHECK-NEXT: vpermi2b %xmm2, %xmm1, %xmm0
269 %res = call <16 x i8> @llvm.x86.avx512.mask.vpermt2var.qi.128(<16 x i8> %x0, <16 x i8> %x1, <16 x i8> %x2, i16 -1)
273 define <16 x i8>@test_int_x86_avx512_mask_vpermt2var_qi_128_load(<16 x i8> %x0, <16 x i8> %x1, ptr %x2p) {
274 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermt2var_qi_128_load:
276 ; CHECK-NEXT: vpermi2b (%rdi), %xmm1, %xmm0
278 %x2 = load <16 x i8>, ptr %x2p
279 %res = call <16 x i8> @llvm.x86.avx512.mask.vpermt2var.qi.128(<16 x i8> %x0, <16 x i8> %x1, <16 x i8> %x2, i16 -1)
283 declare <32 x i8> @llvm.x86.avx512.mask.vpermt2var.qi.256(<32 x i8>, <32 x i8>, <32 x i8>, i32)
285 define <32 x i8>@test_int_x86_avx512_mask_vpermt2var_qi_256(<32 x i8> %x0, <32 x i8> %x1, <32 x i8> %x2) {
286 ; CHECK-LABEL: test_int_x86_avx512_mask_vpermt2var_qi_256:
288 ; CHECK-NEXT: vpermi2b %ymm2, %ymm1, %ymm0
290 %res = call <32 x i8> @llvm.x86.avx512.mask.vpermt2var.qi.256(<32 x i8> %x0, <32 x i8> %x1, <32 x i8> %x2, i32 -1)
294 declare <16 x i8> @llvm.x86.avx512.maskz.vpermt2var.qi.128(<16 x i8>, <16 x i8>, <16 x i8>, i16)
296 define <16 x i8>@test_int_x86_avx512_maskz_vpermt2var_qi_128(<16 x i8> %x0, <16 x i8> %x1, <16 x i8> %x2, i16 %x3) {
297 ; CHECK-LABEL: test_int_x86_avx512_maskz_vpermt2var_qi_128:
299 ; CHECK-NEXT: kmovd %edi, %k1
300 ; CHECK-NEXT: vpermi2b %xmm2, %xmm1, %xmm0 {%k1} {z}
302 %res = call <16 x i8> @llvm.x86.avx512.maskz.vpermt2var.qi.128(<16 x i8> %x0, <16 x i8> %x1, <16 x i8> %x2, i16 %x3)
306 define <16 x i8>@test_int_x86_avx512_maskz_vpermt2var_qi_128_load(<16 x i8> %x0, <16 x i8> %x1, ptr %x2p, i16 %x3) {
307 ; CHECK-LABEL: test_int_x86_avx512_maskz_vpermt2var_qi_128_load:
309 ; CHECK-NEXT: kmovd %esi, %k1
310 ; CHECK-NEXT: vpermi2b (%rdi), %xmm1, %xmm0 {%k1} {z}
312 %x2 = load <16 x i8>, ptr %x2p
313 %res = call <16 x i8> @llvm.x86.avx512.maskz.vpermt2var.qi.128(<16 x i8> %x0, <16 x i8> %x1, <16 x i8> %x2, i16 %x3)
317 declare <32 x i8> @llvm.x86.avx512.maskz.vpermt2var.qi.256(<32 x i8>, <32 x i8>, <32 x i8>, i32)
319 define <32 x i8>@test_int_x86_avx512_maskz_vpermt2var_qi_256(<32 x i8> %x0, <32 x i8> %x1, <32 x i8> %x2, i32 %x3) {
320 ; CHECK-LABEL: test_int_x86_avx512_maskz_vpermt2var_qi_256:
322 ; CHECK-NEXT: kmovd %edi, %k1
323 ; CHECK-NEXT: vpermi2b %ymm2, %ymm1, %ymm0 {%k1} {z}
325 %res = call <32 x i8> @llvm.x86.avx512.maskz.vpermt2var.qi.256(<32 x i8> %x0, <32 x i8> %x1, <32 x i8> %x2, i32 %x3)
329 define <32 x i8>@test_int_x86_avx512_maskz_vpermt2var_qi_256_load(<32 x i8> %x0, <32 x i8> %x1, ptr %x2p, i32 %x3) {
330 ; CHECK-LABEL: test_int_x86_avx512_maskz_vpermt2var_qi_256_load:
332 ; CHECK-NEXT: kmovd %esi, %k1
333 ; CHECK-NEXT: vpermi2b (%rdi), %ymm1, %ymm0 {%k1} {z}
335 %x2 = load <32 x i8>, ptr %x2p
336 %res = call <32 x i8> @llvm.x86.avx512.maskz.vpermt2var.qi.256(<32 x i8> %x0, <32 x i8> %x1, <32 x i8> %x2, i32 %x3)