| blob | 3cd5654771c5a345f327f581872765e4cfbb83db |
1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2 ; RUN: llc < %s -mtriple=x86_64-apple-macosx -mattr=+avx | FileCheck %s --check-prefix=AVX --check-prefix=AVX1
3 ; RUN: llc < %s -mtriple=x86_64-apple-macosx -mattr=+avx2 | FileCheck %s --check-prefix=AVX --check-prefix=AVX2
5 target datalayout = "e-m:o-i64:64-f80:128-n8:16:32:64-S128"
7 ; For this test we used to optimize the <i1 true, i1 false, i1 false, i1 true>
8 ; mask into <i32 2147483648, i32 0, i32 0, i32 2147483648> because we thought
9 ; we would lower that into a blend where only the high bit is relevant.
10 ; However, since the whole mask is constant, this is simplified incorrectly
11 ; by the generic code, because it was expecting -1 in place of 2147483648.
12 ;
13 ; The problem does not occur without AVX, because vselect of v4i32 is not legal
14 ; nor custom.
15 ;
16 ; <rdar://problem/18675020>
18 define void @test(<4 x i16>* %a, <4 x i16>* %b) {
19 ; AVX-LABEL: test:
20 ; AVX: ## %bb.0: ## %body
21 ; AVX-NEXT: movabsq $4167800517033787389, %rax ## imm = 0x39D7007D007CFFFD
22 ; AVX-NEXT: movq %rax, (%rdi)
23 ; AVX-NEXT: movabsq $-281474976645121, %rax ## imm = 0xFFFF00000000FFFF
24 ; AVX-NEXT: movq %rax, (%rsi)
25 ; AVX-NEXT: retq
26 body:
27 %predphi = select <4 x i1> <i1 true, i1 false, i1 false, i1 true>, <4 x i16> <i16 -3, i16 545, i16 4385, i16 14807>, <4 x i16> <i16 123, i16 124, i16 125, i16 127>
28 %predphi42 = select <4 x i1> <i1 true, i1 false, i1 false, i1 true>, <4 x i16> <i16 -1, i16 -1, i16 -1, i16 -1>, <4 x i16> zeroinitializer
29 store <4 x i16> %predphi, <4 x i16>* %a, align 8
30 store <4 x i16> %predphi42, <4 x i16>* %b, align 8
31 ret void
32 }
34 ; Improve code coverage.
35 ;
36 ; When shrinking the condition used into the select to match a blend, this
37 ; test case exercises the path where the modified node is not the root
38 ; of the condition.
40 define void @test2(double** %call1559, i64 %indvars.iv4198, <4 x i1> %tmp1895) {
41 ; AVX1-LABEL: test2:
42 ; AVX1: ## %bb.0: ## %bb
43 ; AVX1-NEXT: vpslld $31, %xmm0, %xmm0
44 ; AVX1-NEXT: vpmovsxdq %xmm0, %xmm1
45 ; AVX1-NEXT: vpshufd {{.*#+}} xmm0 = xmm0[2,3,0,1]
46 ; AVX1-NEXT: vpmovsxdq %xmm0, %xmm0
47 ; AVX1-NEXT: vinsertf128 $1, %xmm0, %ymm1, %ymm0
48 ; AVX1-NEXT: movq (%rdi,%rsi,8), %rax
49 ; AVX1-NEXT: vmovapd {{.*#+}} ymm1 = [5.0E-1,5.0E-1,5.0E-1,5.0E-1]
50 ; AVX1-NEXT: vblendvpd %ymm0, {{.*}}(%rip), %ymm1, %ymm0
51 ; AVX1-NEXT: vmovupd %ymm0, (%rax)
52 ; AVX1-NEXT: vzeroupper
53 ; AVX1-NEXT: retq
54 ;
55 ; AVX2-LABEL: test2:
56 ; AVX2: ## %bb.0: ## %bb
57 ; AVX2-NEXT: vpslld $31, %xmm0, %xmm0
58 ; AVX2-NEXT: vpmovsxdq %xmm0, %ymm0
59 ; AVX2-NEXT: movq (%rdi,%rsi,8), %rax
60 ; AVX2-NEXT: vbroadcastsd {{.*#+}} ymm1 = [-5.0E-1,-5.0E-1,-5.0E-1,-5.0E-1]
61 ; AVX2-NEXT: vbroadcastsd {{.*#+}} ymm2 = [5.0E-1,5.0E-1,5.0E-1,5.0E-1]
62 ; AVX2-NEXT: vblendvpd %ymm0, %ymm1, %ymm2, %ymm0
63 ; AVX2-NEXT: vmovupd %ymm0, (%rax)
64 ; AVX2-NEXT: vzeroupper
65 ; AVX2-NEXT: retq
66 bb:
67 %arrayidx1928 = getelementptr inbounds double*, double** %call1559, i64 %indvars.iv4198
68 %tmp1888 = load double*, double** %arrayidx1928, align 8
69 %predphi.v.v = select <4 x i1> %tmp1895, <4 x double> <double -5.000000e-01, double -5.000000e-01, double -5.000000e-01, double -5.000000e-01>, <4 x double> <double 5.000000e-01, double 5.000000e-01, double 5.000000e-01, double 5.000000e-01>
70 %tmp1900 = bitcast double* %tmp1888 to <4 x double>*
71 store <4 x double> %predphi.v.v, <4 x double>* %tmp1900, align 8
72 ret void
73 }
75 ; For this test, we used to optimized the conditional mask for the blend, i.e.,
76 ; we shrunk some of its bits.
77 ; However, this same mask was used in another select (%predphi31) that turned out
78 ; to be optimized into a and. In that case, the conditional mask was wrong.
79 ;
80 ; Make sure that the and is fed by the original mask.
81 ;
82 ; <rdar://problem/18819506>
84 define void @test3(<4 x i32> %induction30, <4 x i16>* %tmp16, <4 x i16>* %tmp17, <4 x i16> %tmp3, <4 x i16> %tmp12) {
85 ; AVX1-LABEL: test3:
86 ; AVX1: ## %bb.0:
87 ; AVX1-NEXT: vpmulld {{.*}}(%rip), %xmm0, %xmm0
88 ; AVX1-NEXT: vpaddd {{.*}}(%rip), %xmm0, %xmm0
89 ; AVX1-NEXT: vpminud {{.*}}(%rip), %xmm0, %xmm3
90 ; AVX1-NEXT: vpcmpeqd %xmm3, %xmm0, %xmm0
91 ; AVX1-NEXT: vpackssdw %xmm0, %xmm0, %xmm0
92 ; AVX1-NEXT: vpblendvb %xmm0, %xmm1, %xmm2, %xmm1
93 ; AVX1-NEXT: vpcmpeqd %xmm2, %xmm2, %xmm2
94 ; AVX1-NEXT: vpxor %xmm3, %xmm3, %xmm3
95 ; AVX1-NEXT: vpblendvb %xmm0, %xmm2, %xmm3, %xmm0
96 ; AVX1-NEXT: vmovq %xmm0, (%rdi)
97 ; AVX1-NEXT: vmovq %xmm1, (%rsi)
98 ; AVX1-NEXT: retq
99 ;
100 ; AVX2-LABEL: test3:
101 ; AVX2: ## %bb.0:
102 ; AVX2-NEXT: vpbroadcastd {{.*#+}} xmm3 = [2863311531,2863311531,2863311531,2863311531]
103 ; AVX2-NEXT: vpmulld %xmm3, %xmm0, %xmm0
104 ; AVX2-NEXT: vpbroadcastd {{.*#+}} xmm3 = [715827882,715827882,715827882,715827882]
105 ; AVX2-NEXT: vpaddd %xmm3, %xmm0, %xmm0
106 ; AVX2-NEXT: vpbroadcastd {{.*#+}} xmm3 = [1431655764,1431655764,1431655764,1431655764]
107 ; AVX2-NEXT: vpminud %xmm3, %xmm0, %xmm3
108 ; AVX2-NEXT: vpcmpeqd %xmm3, %xmm0, %xmm0
109 ; AVX2-NEXT: vpackssdw %xmm0, %xmm0, %xmm0
110 ; AVX2-NEXT: vpblendvb %xmm0, %xmm1, %xmm2, %xmm1
111 ; AVX2-NEXT: vpcmpeqd %xmm2, %xmm2, %xmm2
112 ; AVX2-NEXT: vpxor %xmm3, %xmm3, %xmm3
113 ; AVX2-NEXT: vpblendvb %xmm0, %xmm2, %xmm3, %xmm0
114 ; AVX2-NEXT: vmovq %xmm0, (%rdi)
115 ; AVX2-NEXT: vmovq %xmm1, (%rsi)
116 ; AVX2-NEXT: retq
117 %tmp6 = srem <4 x i32> %induction30, <i32 3, i32 3, i32 3, i32 3>
118 %tmp7 = icmp eq <4 x i32> %tmp6, zeroinitializer
119 %predphi = select <4 x i1> %tmp7, <4 x i16> %tmp3, <4 x i16> %tmp12
120 %predphi31 = select <4 x i1> %tmp7, <4 x i16> <i16 -1, i16 -1, i16 -1, i16 -1>, <4 x i16> zeroinitializer
122 store <4 x i16> %predphi31, <4 x i16>* %tmp16, align 8
123 store <4 x i16> %predphi, <4 x i16>* %tmp17, align 8
124 ret void
125 }
127 ; We shouldn't try to lower this directly using VSELECT because we don't have
128 ; vpblendvb in AVX1, only in AVX2. Instead, it should be expanded.
130 define <32 x i8> @PR22706(<32 x i1> %x) {
131 ; AVX1-LABEL: PR22706:
132 ; AVX1: ## %bb.0:
133 ; AVX1-NEXT: vextractf128 $1, %ymm0, %xmm1
134 ; AVX1-NEXT: vpsllw $7, %xmm1, %xmm1
135 ; AVX1-NEXT: vmovdqa {{.*#+}} xmm2 = [128,128,128,128,128,128,128,128,128,128,128,128,128,128,128,128]
136 ; AVX1-NEXT: vpand %xmm2, %xmm1, %xmm1
137 ; AVX1-NEXT: vpxor %xmm3, %xmm3, %xmm3
138 ; AVX1-NEXT: vpcmpgtb %xmm1, %xmm3, %xmm1
139 ; AVX1-NEXT: vmovdqa {{.*#+}} xmm4 = [2,2,2,2,2,2,2,2,2,2,2,2,2,2,2,2]
140 ; AVX1-NEXT: vpaddb %xmm4, %xmm1, %xmm1
141 ; AVX1-NEXT: vpsllw $7, %xmm0, %xmm0
142 ; AVX1-NEXT: vpand %xmm2, %xmm0, %xmm0
143 ; AVX1-NEXT: vpcmpgtb %xmm0, %xmm3, %xmm0
144 ; AVX1-NEXT: vpaddb %xmm4, %xmm0, %xmm0
145 ; AVX1-NEXT: vinsertf128 $1, %xmm1, %ymm0, %ymm0
146 ; AVX1-NEXT: retq
147 ;
148 ; AVX2-LABEL: PR22706:
149 ; AVX2: ## %bb.0:
150 ; AVX2-NEXT: vpsllw $7, %ymm0, %ymm0
151 ; AVX2-NEXT: vpand {{.*}}(%rip), %ymm0, %ymm0
152 ; AVX2-NEXT: vpxor %xmm1, %xmm1, %xmm1
153 ; AVX2-NEXT: vpcmpgtb %ymm0, %ymm1, %ymm0
154 ; AVX2-NEXT: vpaddb {{.*}}(%rip), %ymm0, %ymm0
155 ; AVX2-NEXT: retq
156 %tmp = select <32 x i1> %x, <32 x i8> <i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1, i8 1>, <32 x i8> <i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2, i8 2>
157 ret <32 x i8> %tmp
158 }
160 ; Split a 256-bit select into two 128-bit selects when the operands are concatenated.
162 define void @blendv_split(<8 x i32>* %p, <8 x i32> %cond, <8 x i32> %a, <8 x i32> %x, <8 x i32> %y, <8 x i32> %z, <8 x i32> %w) {
163 ; AVX1-LABEL: blendv_split:
164 ; AVX1: ## %bb.0:
165 ; AVX1-NEXT: vpmovzxdq {{.*#+}} xmm2 = xmm2[0],zero,xmm2[1],zero
166 ; AVX1-NEXT: vpmovzxdq {{.*#+}} xmm3 = xmm3[0],zero,xmm3[1],zero
167 ; AVX1-NEXT: vextractf128 $1, %ymm1, %xmm4
168 ; AVX1-NEXT: vpslld %xmm2, %xmm4, %xmm5
169 ; AVX1-NEXT: vpslld %xmm2, %xmm1, %xmm2
170 ; AVX1-NEXT: vpslld %xmm3, %xmm4, %xmm4
171 ; AVX1-NEXT: vpslld %xmm3, %xmm1, %xmm1
172 ; AVX1-NEXT: vblendvps %xmm0, %xmm2, %xmm1, %xmm1
173 ; AVX1-NEXT: vextractf128 $1, %ymm0, %xmm0
174 ; AVX1-NEXT: vblendvps %xmm0, %xmm5, %xmm4, %xmm0
175 ; AVX1-NEXT: vmovups %xmm0, 16(%rdi)
176 ; AVX1-NEXT: vmovups %xmm1, (%rdi)
177 ; AVX1-NEXT: vzeroupper
178 ; AVX1-NEXT: retq
179 ;
180 ; AVX2-LABEL: blendv_split:
181 ; AVX2: ## %bb.0:
182 ; AVX2-NEXT: vpmovzxdq {{.*#+}} xmm2 = xmm2[0],zero,xmm2[1],zero
183 ; AVX2-NEXT: vpmovzxdq {{.*#+}} xmm3 = xmm3[0],zero,xmm3[1],zero
184 ; AVX2-NEXT: vpslld %xmm2, %ymm1, %ymm2
185 ; AVX2-NEXT: vpslld %xmm3, %ymm1, %ymm1
186 ; AVX2-NEXT: vblendvps %ymm0, %ymm2, %ymm1, %ymm0
187 ; AVX2-NEXT: vmovups %ymm0, (%rdi)
188 ; AVX2-NEXT: vzeroupper
189 ; AVX2-NEXT: retq
190 %signbits = ashr <8 x i32> %cond, <i32 31, i32 31, i32 31, i32 31, i32 31, i32 31, i32 31, i32 31>
191 %bool = trunc <8 x i32> %signbits to <8 x i1>
192 %shamt1 = shufflevector <8 x i32> %x, <8 x i32> undef, <8 x i32> zeroinitializer
193 %shamt2 = shufflevector <8 x i32> %y, <8 x i32> undef, <8 x i32> zeroinitializer
194 %sh1 = shl <8 x i32> %a, %shamt1
195 %sh2 = shl <8 x i32> %a, %shamt2
196 %sel = select <8 x i1> %bool, <8 x i32> %sh1, <8 x i32> %sh2
197 store <8 x i32> %sel, <8 x i32>* %p, align 4
198 ret void
199 }