llvm/test/CodeGen/X86/cast-vsel.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
   2 ; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+sse2 | FileCheck %s --check-prefix=SSE2
   3 ; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+sse4.1 | FileCheck %s --check-prefix=SSE41
   4 ; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+avx | FileCheck %s --check-prefixes=AVX,AVX1
   5 ; RUN: llc < %s -mtriple=x86_64-unknown-unknown -mattr=+avx2 | FileCheck %s --check-prefixes=AVX,AVX2
   6
   7 ; If we have a cmp and a sel with different-sized operands followed by a size-changing cast,
   8 ; we may want to pull the cast ahead of the select operands to create a select with matching op sizes:
   9 ; ext (sel (cmp a, b), c, d) --> sel (cmp a, b), (ext c), (ext d)
  10
  11 define <8 x i32> @sext(<8 x float> %a, <8 x float> %b, <8 x i16> %c, <8 x i16> %d) {
  12 ; SSE2-LABEL: sext:
  13 ; SSE2:       # %bb.0:
  14 ; SSE2-NEXT:    cmpltps %xmm3, %xmm1
  15 ; SSE2-NEXT:    cmpltps %xmm2, %xmm0
  16 ; SSE2-NEXT:    packssdw %xmm1, %xmm0
  17 ; SSE2-NEXT:    pand %xmm0, %xmm4
  18 ; SSE2-NEXT:    pandn %xmm5, %xmm0
  19 ; SSE2-NEXT:    por %xmm4, %xmm0
  20 ; SSE2-NEXT:    punpcklwd {{.*#+}} xmm2 = xmm2[0],xmm0[0],xmm2[1],xmm0[1],xmm2[2],xmm0[2],xmm2[3],xmm0[3]
  21 ; SSE2-NEXT:    psrad $16, %xmm2
  22 ; SSE2-NEXT:    punpckhwd {{.*#+}} xmm1 = xmm1[4],xmm0[4],xmm1[5],xmm0[5],xmm1[6],xmm0[6],xmm1[7],xmm0[7]
  23 ; SSE2-NEXT:    psrad $16, %xmm1
  24 ; SSE2-NEXT:    movdqa %xmm2, %xmm0
  25 ; SSE2-NEXT:    retq
  26 ;
  27 ; SSE41-LABEL: sext:
  28 ; SSE41:       # %bb.0:
  29 ; SSE41-NEXT:    cmpltps %xmm3, %xmm1
  30 ; SSE41-NEXT:    cmpltps %xmm2, %xmm0
  31 ; SSE41-NEXT:    packssdw %xmm1, %xmm0
  32 ; SSE41-NEXT:    pblendvb %xmm0, %xmm4, %xmm5
  33 ; SSE41-NEXT:    pmovsxwd %xmm5, %xmm0
  34 ; SSE41-NEXT:    pshufd {{.*#+}} xmm1 = xmm5[2,3,2,3]
  35 ; SSE41-NEXT:    pmovsxwd %xmm1, %xmm1
  36 ; SSE41-NEXT:    retq
  37 ;
  38 ; AVX1-LABEL: sext:
  39 ; AVX1:       # %bb.0:
  40 ; AVX1-NEXT:    vcmpltps %ymm1, %ymm0, %ymm0
  41 ; AVX1-NEXT:    vextractf128 $1, %ymm0, %xmm1
  42 ; AVX1-NEXT:    vpshufd {{.*#+}} xmm4 = xmm2[2,3,2,3]
  43 ; AVX1-NEXT:    vpmovsxwd %xmm4, %xmm4
  44 ; AVX1-NEXT:    vpshufd {{.*#+}} xmm5 = xmm3[2,3,2,3]
  45 ; AVX1-NEXT:    vpmovsxwd %xmm5, %xmm5
  46 ; AVX1-NEXT:    vblendvps %xmm1, %xmm4, %xmm5, %xmm1
  47 ; AVX1-NEXT:    vpmovsxwd %xmm2, %xmm2
  48 ; AVX1-NEXT:    vpmovsxwd %xmm3, %xmm3
  49 ; AVX1-NEXT:    vblendvps %xmm0, %xmm2, %xmm3, %xmm0
  50 ; AVX1-NEXT:    vinsertf128 $1, %xmm1, %ymm0, %ymm0
  51 ; AVX1-NEXT:    retq
  52 ;
  53 ; AVX2-LABEL: sext:
  54 ; AVX2:       # %bb.0:
  55 ; AVX2-NEXT:    vcmpltps %ymm1, %ymm0, %ymm0
  56 ; AVX2-NEXT:    vpmovsxwd %xmm2, %ymm1
  57 ; AVX2-NEXT:    vpmovsxwd %xmm3, %ymm2
  58 ; AVX2-NEXT:    vblendvps %ymm0, %ymm1, %ymm2, %ymm0
  59 ; AVX2-NEXT:    retq
  60   %cmp = fcmp olt <8 x float> %a, %b
  61   %sel = select <8 x i1> %cmp, <8 x i16> %c, <8 x i16> %d
  62   %ext = sext <8 x i16> %sel to <8 x i32>
  63   ret <8 x i32> %ext
  64 }
  65
  66 define <8 x i32> @zext(<8 x float> %a, <8 x float> %b, <8 x i16> %c, <8 x i16> %d) {
  67 ; SSE2-LABEL: zext:
  68 ; SSE2:       # %bb.0:
  69 ; SSE2-NEXT:    movaps %xmm0, %xmm6
  70 ; SSE2-NEXT:    cmpltps %xmm3, %xmm1
  71 ; SSE2-NEXT:    cmpltps %xmm2, %xmm6
  72 ; SSE2-NEXT:    packssdw %xmm1, %xmm6
  73 ; SSE2-NEXT:    pand %xmm6, %xmm4
  74 ; SSE2-NEXT:    pandn %xmm5, %xmm6
  75 ; SSE2-NEXT:    por %xmm4, %xmm6
  76 ; SSE2-NEXT:    pxor %xmm1, %xmm1
  77 ; SSE2-NEXT:    movdqa %xmm6, %xmm0
  78 ; SSE2-NEXT:    punpcklwd {{.*#+}} xmm0 = xmm0[0],xmm1[0],xmm0[1],xmm1[1],xmm0[2],xmm1[2],xmm0[3],xmm1[3]
  79 ; SSE2-NEXT:    punpckhwd {{.*#+}} xmm6 = xmm6[4],xmm1[4],xmm6[5],xmm1[5],xmm6[6],xmm1[6],xmm6[7],xmm1[7]
  80 ; SSE2-NEXT:    movdqa %xmm6, %xmm1
  81 ; SSE2-NEXT:    retq
  82 ;
  83 ; SSE41-LABEL: zext:
  84 ; SSE41:       # %bb.0:
  85 ; SSE41-NEXT:    cmpltps %xmm3, %xmm1
  86 ; SSE41-NEXT:    cmpltps %xmm2, %xmm0
  87 ; SSE41-NEXT:    packssdw %xmm1, %xmm0
  88 ; SSE41-NEXT:    pblendvb %xmm0, %xmm4, %xmm5
  89 ; SSE41-NEXT:    pxor %xmm1, %xmm1
  90 ; SSE41-NEXT:    pmovzxwd {{.*#+}} xmm0 = xmm5[0],zero,xmm5[1],zero,xmm5[2],zero,xmm5[3],zero
  91 ; SSE41-NEXT:    punpckhwd {{.*#+}} xmm5 = xmm5[4],xmm1[4],xmm5[5],xmm1[5],xmm5[6],xmm1[6],xmm5[7],xmm1[7]
  92 ; SSE41-NEXT:    movdqa %xmm5, %xmm1
  93 ; SSE41-NEXT:    retq
  94 ;
  95 ; AVX1-LABEL: zext:
  96 ; AVX1:       # %bb.0:
  97 ; AVX1-NEXT:    vcmpltps %ymm1, %ymm0, %ymm0
  98 ; AVX1-NEXT:    vxorps %xmm1, %xmm1, %xmm1
  99 ; AVX1-NEXT:    vpunpckhwd {{.*#+}} xmm4 = xmm2[4],xmm1[4],xmm2[5],xmm1[5],xmm2[6],xmm1[6],xmm2[7],xmm1[7]
 100 ; AVX1-NEXT:    vpunpckhwd {{.*#+}} xmm1 = xmm3[4],xmm1[4],xmm3[5],xmm1[5],xmm3[6],xmm1[6],xmm3[7],xmm1[7]
 101 ; AVX1-NEXT:    vextractf128 $1, %ymm0, %xmm5
 102 ; AVX1-NEXT:    vblendvps %xmm5, %xmm4, %xmm1, %xmm1
 103 ; AVX1-NEXT:    vpmovzxwd {{.*#+}} xmm2 = xmm2[0],zero,xmm2[1],zero,xmm2[2],zero,xmm2[3],zero
 104 ; AVX1-NEXT:    vpmovzxwd {{.*#+}} xmm3 = xmm3[0],zero,xmm3[1],zero,xmm3[2],zero,xmm3[3],zero
 105 ; AVX1-NEXT:    vblendvps %xmm0, %xmm2, %xmm3, %xmm0
 106 ; AVX1-NEXT:    vinsertf128 $1, %xmm1, %ymm0, %ymm0
 107 ; AVX1-NEXT:    retq
 108 ;
 109 ; AVX2-LABEL: zext:
 110 ; AVX2:       # %bb.0:
 111 ; AVX2-NEXT:    vcmpltps %ymm1, %ymm0, %ymm0
 112 ; AVX2-NEXT:    vpmovzxwd {{.*#+}} ymm1 = xmm2[0],zero,xmm2[1],zero,xmm2[2],zero,xmm2[3],zero,xmm2[4],zero,xmm2[5],zero,xmm2[6],zero,xmm2[7],zero
 113 ; AVX2-NEXT:    vpmovzxwd {{.*#+}} ymm2 = xmm3[0],zero,xmm3[1],zero,xmm3[2],zero,xmm3[3],zero,xmm3[4],zero,xmm3[5],zero,xmm3[6],zero,xmm3[7],zero
 114 ; AVX2-NEXT:    vblendvps %ymm0, %ymm1, %ymm2, %ymm0
 115 ; AVX2-NEXT:    retq
 116   %cmp = fcmp olt <8 x float> %a, %b
 117   %sel = select <8 x i1> %cmp, <8 x i16> %c, <8 x i16> %d
 118   %ext = zext <8 x i16> %sel to <8 x i32>
 119   ret <8 x i32> %ext
 120 }
 121
 122 define <4 x double> @fpext(<4 x double> %a, <4 x double> %b, <4 x float> %c, <4 x float> %d) {
 123 ; SSE2-LABEL: fpext:
 124 ; SSE2:       # %bb.0:
 125 ; SSE2-NEXT:    cmpltpd %xmm3, %xmm1
 126 ; SSE2-NEXT:    cmpltpd %xmm2, %xmm0
 127 ; SSE2-NEXT:    shufps {{.*#+}} xmm0 = xmm0[0,2],xmm1[0,2]
 128 ; SSE2-NEXT:    andps %xmm0, %xmm4
 129 ; SSE2-NEXT:    andnps %xmm5, %xmm0
 130 ; SSE2-NEXT:    orps %xmm4, %xmm0
 131 ; SSE2-NEXT:    cvtps2pd %xmm0, %xmm2
 132 ; SSE2-NEXT:    pshufd {{.*#+}} xmm0 = xmm0[2,3,2,3]
 133 ; SSE2-NEXT:    cvtps2pd %xmm0, %xmm1
 134 ; SSE2-NEXT:    movaps %xmm2, %xmm0
 135 ; SSE2-NEXT:    retq
 136 ;
 137 ; SSE41-LABEL: fpext:
 138 ; SSE41:       # %bb.0:
 139 ; SSE41-NEXT:    cmpltpd %xmm3, %xmm1
 140 ; SSE41-NEXT:    cmpltpd %xmm2, %xmm0
 141 ; SSE41-NEXT:    shufps {{.*#+}} xmm0 = xmm0[0,2],xmm1[0,2]
 142 ; SSE41-NEXT:    blendvps %xmm0, %xmm4, %xmm5
 143 ; SSE41-NEXT:    cvtps2pd %xmm5, %xmm0
 144 ; SSE41-NEXT:    movhlps {{.*#+}} xmm5 = xmm5[1,1]
 145 ; SSE41-NEXT:    cvtps2pd %xmm5, %xmm1
 146 ; SSE41-NEXT:    retq
 147 ;
 148 ; AVX-LABEL: fpext:
 149 ; AVX:       # %bb.0:
 150 ; AVX-NEXT:    vcmpltpd %ymm1, %ymm0, %ymm0
 151 ; AVX-NEXT:    vcvtps2pd %xmm2, %ymm1
 152 ; AVX-NEXT:    vcvtps2pd %xmm3, %ymm2
 153 ; AVX-NEXT:    vblendvpd %ymm0, %ymm1, %ymm2, %ymm0
 154 ; AVX-NEXT:    retq
 155   %cmp = fcmp olt <4 x double> %a, %b
 156   %sel = select <4 x i1> %cmp, <4 x float> %c, <4 x float> %d
 157   %ext = fpext <4 x float> %sel to <4 x double>
 158   ret <4 x double> %ext
 159 }
 160
 161 define <8 x i16> @trunc(<8 x i16> %a, <8 x i16> %b, <8 x i32> %c, <8 x i32> %d) {
 162 ; SSE2-LABEL: trunc:
 163 ; SSE2:       # %bb.0:
 164 ; SSE2-NEXT:    pcmpeqw %xmm1, %xmm0
 165 ; SSE2-NEXT:    pslld $16, %xmm5
 166 ; SSE2-NEXT:    psrad $16, %xmm5
 167 ; SSE2-NEXT:    pslld $16, %xmm4
 168 ; SSE2-NEXT:    psrad $16, %xmm4
 169 ; SSE2-NEXT:    packssdw %xmm5, %xmm4
 170 ; SSE2-NEXT:    pslld $16, %xmm3
 171 ; SSE2-NEXT:    psrad $16, %xmm3
 172 ; SSE2-NEXT:    pslld $16, %xmm2
 173 ; SSE2-NEXT:    psrad $16, %xmm2
 174 ; SSE2-NEXT:    packssdw %xmm3, %xmm2
 175 ; SSE2-NEXT:    pand %xmm0, %xmm2
 176 ; SSE2-NEXT:    pandn %xmm4, %xmm0
 177 ; SSE2-NEXT:    por %xmm2, %xmm0
 178 ; SSE2-NEXT:    retq
 179 ;
 180 ; SSE41-LABEL: trunc:
 181 ; SSE41:       # %bb.0:
 182 ; SSE41-NEXT:    pcmpeqw %xmm1, %xmm0
 183 ; SSE41-NEXT:    pxor %xmm1, %xmm1
 184 ; SSE41-NEXT:    pblendw {{.*#+}} xmm3 = xmm3[0],xmm1[1],xmm3[2],xmm1[3],xmm3[4],xmm1[5],xmm3[6],xmm1[7]
 185 ; SSE41-NEXT:    pblendw {{.*#+}} xmm2 = xmm2[0],xmm1[1],xmm2[2],xmm1[3],xmm2[4],xmm1[5],xmm2[6],xmm1[7]
 186 ; SSE41-NEXT:    packusdw %xmm3, %xmm2
 187 ; SSE41-NEXT:    pblendw {{.*#+}} xmm5 = xmm5[0],xmm1[1],xmm5[2],xmm1[3],xmm5[4],xmm1[5],xmm5[6],xmm1[7]
 188 ; SSE41-NEXT:    pblendw {{.*#+}} xmm1 = xmm4[0],xmm1[1],xmm4[2],xmm1[3],xmm4[4],xmm1[5],xmm4[6],xmm1[7]
 189 ; SSE41-NEXT:    packusdw %xmm5, %xmm1
 190 ; SSE41-NEXT:    pblendvb %xmm0, %xmm2, %xmm1
 191 ; SSE41-NEXT:    movdqa %xmm1, %xmm0
 192 ; SSE41-NEXT:    retq
 193 ;
 194 ; AVX1-LABEL: trunc:
 195 ; AVX1:       # %bb.0:
 196 ; AVX1-NEXT:    vpcmpeqw %xmm1, %xmm0, %xmm0
 197 ; AVX1-NEXT:    vbroadcastss {{.*#+}} ymm1 = [65535,65535,65535,65535,65535,65535,65535,65535]
 198 ; AVX1-NEXT:    vandps %ymm1, %ymm2, %ymm2
 199 ; AVX1-NEXT:    vextractf128 $1, %ymm2, %xmm4
 200 ; AVX1-NEXT:    vpackusdw %xmm4, %xmm2, %xmm2
 201 ; AVX1-NEXT:    vandps %ymm1, %ymm3, %ymm1
 202 ; AVX1-NEXT:    vextractf128 $1, %ymm1, %xmm3
 203 ; AVX1-NEXT:    vpackusdw %xmm3, %xmm1, %xmm1
 204 ; AVX1-NEXT:    vpblendvb %xmm0, %xmm2, %xmm1, %xmm0
 205 ; AVX1-NEXT:    vzeroupper
 206 ; AVX1-NEXT:    retq
 207 ;
 208 ; AVX2-LABEL: trunc:
 209 ; AVX2:       # %bb.0:
 210 ; AVX2-NEXT:    vpcmpeqw %xmm1, %xmm0, %xmm0
 211 ; AVX2-NEXT:    vmovdqa {{.*#+}} ymm1 = [0,1,4,5,8,9,12,13,8,9,12,13,12,13,14,15,16,17,20,21,24,25,28,29,24,25,28,29,28,29,30,31]
 212 ; AVX2-NEXT:    vpshufb %ymm1, %ymm2, %ymm2
 213 ; AVX2-NEXT:    vpermq {{.*#+}} ymm2 = ymm2[0,2,2,3]
 214 ; AVX2-NEXT:    vpshufb %ymm1, %ymm3, %ymm1
 215 ; AVX2-NEXT:    vpermq {{.*#+}} ymm1 = ymm1[0,2,2,3]
 216 ; AVX2-NEXT:    vpblendvb %xmm0, %xmm2, %xmm1, %xmm0
 217 ; AVX2-NEXT:    vzeroupper
 218 ; AVX2-NEXT:    retq
 219   %cmp = icmp eq <8 x i16> %a, %b
 220   %sel = select <8 x i1> %cmp, <8 x i32> %c, <8 x i32> %d
 221   %tr = trunc <8 x i32> %sel to <8 x i16>
 222   ret <8 x i16> %tr
 223 }
 224
 225 define <4 x float> @fptrunc(<4 x float> %a, <4 x float> %b, <4 x double> %c, <4 x double> %d) {
 226 ; SSE2-LABEL: fptrunc:
 227 ; SSE2:       # %bb.0:
 228 ; SSE2-NEXT:    cmpltps %xmm1, %xmm0
 229 ; SSE2-NEXT:    cvtpd2ps %xmm5, %xmm1
 230 ; SSE2-NEXT:    cvtpd2ps %xmm4, %xmm4
 231 ; SSE2-NEXT:    unpcklpd {{.*#+}} xmm4 = xmm4[0],xmm1[0]
 232 ; SSE2-NEXT:    cvtpd2ps %xmm3, %xmm1
 233 ; SSE2-NEXT:    cvtpd2ps %xmm2, %xmm2
 234 ; SSE2-NEXT:    unpcklpd {{.*#+}} xmm2 = xmm2[0],xmm1[0]
 235 ; SSE2-NEXT:    andpd %xmm0, %xmm2
 236 ; SSE2-NEXT:    andnpd %xmm4, %xmm0
 237 ; SSE2-NEXT:    orpd %xmm2, %xmm0
 238 ; SSE2-NEXT:    retq
 239 ;
 240 ; SSE41-LABEL: fptrunc:
 241 ; SSE41:       # %bb.0:
 242 ; SSE41-NEXT:    cmpltps %xmm1, %xmm0
 243 ; SSE41-NEXT:    cvtpd2ps %xmm3, %xmm1
 244 ; SSE41-NEXT:    cvtpd2ps %xmm2, %xmm2
 245 ; SSE41-NEXT:    unpcklpd {{.*#+}} xmm2 = xmm2[0],xmm1[0]
 246 ; SSE41-NEXT:    cvtpd2ps %xmm5, %xmm3
 247 ; SSE41-NEXT:    cvtpd2ps %xmm4, %xmm1
 248 ; SSE41-NEXT:    unpcklpd {{.*#+}} xmm1 = xmm1[0],xmm3[0]
 249 ; SSE41-NEXT:    blendvps %xmm0, %xmm2, %xmm1
 250 ; SSE41-NEXT:    movaps %xmm1, %xmm0
 251 ; SSE41-NEXT:    retq
 252 ;
 253 ; AVX-LABEL: fptrunc:
 254 ; AVX:       # %bb.0:
 255 ; AVX-NEXT:    vcmpltps %xmm1, %xmm0, %xmm0
 256 ; AVX-NEXT:    vcvtpd2ps %ymm2, %xmm1
 257 ; AVX-NEXT:    vcvtpd2ps %ymm3, %xmm2
 258 ; AVX-NEXT:    vblendvps %xmm0, %xmm1, %xmm2, %xmm0
 259 ; AVX-NEXT:    vzeroupper
 260 ; AVX-NEXT:    retq
 261   %cmp = fcmp olt <4 x float> %a, %b
 262   %sel = select <4 x i1> %cmp, <4 x double> %c, <4 x double> %d
 263   %tr = fptrunc <4 x double> %sel to <4 x float>
 264   ret <4 x float> %tr
 265 }
 266
 267 ; PR14657 - avoid truncation/extension of comparison results
 268 ; These tests demonstrate the same issue as the simpler cases above,
 269 ; but also include multi-BB to show potentially larger transforms/codegen issues.
 270
 271 @da = common dso_local global [1024 x float] zeroinitializer, align 32
 272 @db = common dso_local global [1024 x float] zeroinitializer, align 32
 273 @dc = common dso_local global [1024 x float] zeroinitializer, align 32
 274 @dd = common dso_local global [1024 x float] zeroinitializer, align 32
 275 @dj = common dso_local global [1024 x i32] zeroinitializer, align 32
 276
 277 define dso_local void @example25() nounwind {
 278 ; SSE2-LABEL: example25:
 279 ; SSE2:       # %bb.0: # %vector.ph
 280 ; SSE2-NEXT:    movq $-4096, %rax # imm = 0xF000
 281 ; SSE2-NEXT:    .p2align 4
 282 ; SSE2-NEXT:  .LBB5_1: # %vector.body
 283 ; SSE2-NEXT:    # =>This Inner Loop Header: Depth=1
 284 ; SSE2-NEXT:    movaps da+4112(%rax), %xmm0
 285 ; SSE2-NEXT:    movaps da+4096(%rax), %xmm1
 286 ; SSE2-NEXT:    cmpltps db+4096(%rax), %xmm1
 287 ; SSE2-NEXT:    cmpltps db+4112(%rax), %xmm0
 288 ; SSE2-NEXT:    movaps dc+4112(%rax), %xmm2
 289 ; SSE2-NEXT:    movaps dc+4096(%rax), %xmm3
 290 ; SSE2-NEXT:    cmpltps dd+4096(%rax), %xmm3
 291 ; SSE2-NEXT:    andps %xmm1, %xmm3
 292 ; SSE2-NEXT:    cmpltps dd+4112(%rax), %xmm2
 293 ; SSE2-NEXT:    andps %xmm0, %xmm2
 294 ; SSE2-NEXT:    psrld $31, %xmm3
 295 ; SSE2-NEXT:    psrld $31, %xmm2
 296 ; SSE2-NEXT:    movdqa %xmm2, dj+4112(%rax)
 297 ; SSE2-NEXT:    movdqa %xmm3, dj+4096(%rax)
 298 ; SSE2-NEXT:    addq $32, %rax
 299 ; SSE2-NEXT:    jne .LBB5_1
 300 ; SSE2-NEXT:  # %bb.2: # %for.end
 301 ; SSE2-NEXT:    retq
 302 ;
 303 ; SSE41-LABEL: example25:
 304 ; SSE41:       # %bb.0: # %vector.ph
 305 ; SSE41-NEXT:    movq $-4096, %rax # imm = 0xF000
 306 ; SSE41-NEXT:    .p2align 4
 307 ; SSE41-NEXT:  .LBB5_1: # %vector.body
 308 ; SSE41-NEXT:    # =>This Inner Loop Header: Depth=1
 309 ; SSE41-NEXT:    movaps da+4112(%rax), %xmm0
 310 ; SSE41-NEXT:    movaps da+4096(%rax), %xmm1
 311 ; SSE41-NEXT:    cmpltps db+4096(%rax), %xmm1
 312 ; SSE41-NEXT:    cmpltps db+4112(%rax), %xmm0
 313 ; SSE41-NEXT:    movaps dc+4112(%rax), %xmm2
 314 ; SSE41-NEXT:    movaps dc+4096(%rax), %xmm3
 315 ; SSE41-NEXT:    cmpltps dd+4096(%rax), %xmm3
 316 ; SSE41-NEXT:    andps %xmm1, %xmm3
 317 ; SSE41-NEXT:    cmpltps dd+4112(%rax), %xmm2
 318 ; SSE41-NEXT:    andps %xmm0, %xmm2
 319 ; SSE41-NEXT:    psrld $31, %xmm3
 320 ; SSE41-NEXT:    psrld $31, %xmm2
 321 ; SSE41-NEXT:    movdqa %xmm2, dj+4112(%rax)
 322 ; SSE41-NEXT:    movdqa %xmm3, dj+4096(%rax)
 323 ; SSE41-NEXT:    addq $32, %rax
 324 ; SSE41-NEXT:    jne .LBB5_1
 325 ; SSE41-NEXT:  # %bb.2: # %for.end
 326 ; SSE41-NEXT:    retq
 327 ;
 328 ; AVX1-LABEL: example25:
 329 ; AVX1:       # %bb.0: # %vector.ph
 330 ; AVX1-NEXT:    movq $-4096, %rax # imm = 0xF000
 331 ; AVX1-NEXT:    vbroadcastss {{.*#+}} ymm0 = [1,1,1,1,1,1,1,1]
 332 ; AVX1-NEXT:    .p2align 4
 333 ; AVX1-NEXT:  .LBB5_1: # %vector.body
 334 ; AVX1-NEXT:    # =>This Inner Loop Header: Depth=1
 335 ; AVX1-NEXT:    vmovups da+4096(%rax), %ymm1
 336 ; AVX1-NEXT:    vcmpltps db+4096(%rax), %ymm1, %ymm1
 337 ; AVX1-NEXT:    vmovups dc+4096(%rax), %ymm2
 338 ; AVX1-NEXT:    vcmpltps dd+4096(%rax), %ymm2, %ymm2
 339 ; AVX1-NEXT:    vandps %ymm2, %ymm1, %ymm1
 340 ; AVX1-NEXT:    vandps %ymm0, %ymm1, %ymm1
 341 ; AVX1-NEXT:    vmovups %ymm1, dj+4096(%rax)
 342 ; AVX1-NEXT:    addq $32, %rax
 343 ; AVX1-NEXT:    jne .LBB5_1
 344 ; AVX1-NEXT:  # %bb.2: # %for.end
 345 ; AVX1-NEXT:    vzeroupper
 346 ; AVX1-NEXT:    retq
 347 ;
 348 ; AVX2-LABEL: example25:
 349 ; AVX2:       # %bb.0: # %vector.ph
 350 ; AVX2-NEXT:    movq $-4096, %rax # imm = 0xF000
 351 ; AVX2-NEXT:    .p2align 4
 352 ; AVX2-NEXT:  .LBB5_1: # %vector.body
 353 ; AVX2-NEXT:    # =>This Inner Loop Header: Depth=1
 354 ; AVX2-NEXT:    vmovups da+4096(%rax), %ymm0
 355 ; AVX2-NEXT:    vcmpltps db+4096(%rax), %ymm0, %ymm0
 356 ; AVX2-NEXT:    vmovups dc+4096(%rax), %ymm1
 357 ; AVX2-NEXT:    vcmpltps dd+4096(%rax), %ymm1, %ymm1
 358 ; AVX2-NEXT:    vandps %ymm1, %ymm0, %ymm0
 359 ; AVX2-NEXT:    vpsrld $31, %ymm0, %ymm0
 360 ; AVX2-NEXT:    vmovdqu %ymm0, dj+4096(%rax)
 361 ; AVX2-NEXT:    addq $32, %rax
 362 ; AVX2-NEXT:    jne .LBB5_1
 363 ; AVX2-NEXT:  # %bb.2: # %for.end
 364 ; AVX2-NEXT:    vzeroupper
 365 ; AVX2-NEXT:    retq
 366 vector.ph:
 367   br label %vector.body
 368
 369 vector.body:
 370   %index = phi i64 [ 0, %vector.ph ], [ %index.next, %vector.body ]
 371   %0 = getelementptr inbounds [1024 x float], ptr @da, i64 0, i64 %index
 372   %1 = load <8 x float>, ptr %0, align 16
 373   %2 = getelementptr inbounds [1024 x float], ptr @db, i64 0, i64 %index
 374   %3 = load <8 x float>, ptr %2, align 16
 375   %4 = fcmp olt <8 x float> %1, %3
 376   %5 = getelementptr inbounds [1024 x float], ptr @dc, i64 0, i64 %index
 377   %6 = load <8 x float>, ptr %5, align 16
 378   %7 = getelementptr inbounds [1024 x float], ptr @dd, i64 0, i64 %index
 379   %8 = load <8 x float>, ptr %7, align 16
 380   %9 = fcmp olt <8 x float> %6, %8
 381   %10 = and <8 x i1> %4, %9
 382   %11 = zext <8 x i1> %10 to <8 x i32>
 383   %12 = getelementptr inbounds [1024 x i32], ptr @dj, i64 0, i64 %index
 384   store <8 x i32> %11, ptr %12, align 16
 385   %index.next = add i64 %index, 8
 386   %13 = icmp eq i64 %index.next, 1024
 387   br i1 %13, label %for.end, label %vector.body
 388
 389 for.end:
 390   ret void
 391 }
 392
 393 ; TODO: AVX1 could have used 256-bit ops for a likely improvement.
 394
 395 define dso_local void @example24(i16 signext %x, i16 signext %y) nounwind {
 396 ; SSE2-LABEL: example24:
 397 ; SSE2:       # %bb.0: # %vector.ph
 398 ; SSE2-NEXT:    movd %edi, %xmm0
 399 ; SSE2-NEXT:    pshuflw {{.*#+}} xmm0 = xmm0[0,0,0,0,4,5,6,7]
 400 ; SSE2-NEXT:    pshufd {{.*#+}} xmm0 = xmm0[0,0,0,0]
 401 ; SSE2-NEXT:    movd %esi, %xmm1
 402 ; SSE2-NEXT:    pshuflw {{.*#+}} xmm1 = xmm1[0,0,0,0,4,5,6,7]
 403 ; SSE2-NEXT:    pshufd {{.*#+}} xmm1 = xmm1[0,0,0,0]
 404 ; SSE2-NEXT:    movq $-4096, %rax # imm = 0xF000
 405 ; SSE2-NEXT:    .p2align 4
 406 ; SSE2-NEXT:  .LBB6_1: # %vector.body
 407 ; SSE2-NEXT:    # =>This Inner Loop Header: Depth=1
 408 ; SSE2-NEXT:    movaps da+4096(%rax), %xmm2
 409 ; SSE2-NEXT:    movaps da+4112(%rax), %xmm3
 410 ; SSE2-NEXT:    cmpltps db+4112(%rax), %xmm3
 411 ; SSE2-NEXT:    cmpltps db+4096(%rax), %xmm2
 412 ; SSE2-NEXT:    packssdw %xmm3, %xmm2
 413 ; SSE2-NEXT:    movdqa %xmm0, %xmm3
 414 ; SSE2-NEXT:    pand %xmm2, %xmm3
 415 ; SSE2-NEXT:    pandn %xmm1, %xmm2
 416 ; SSE2-NEXT:    por %xmm3, %xmm2
 417 ; SSE2-NEXT:    punpcklwd {{.*#+}} xmm3 = xmm3[0],xmm2[0],xmm3[1],xmm2[1],xmm3[2],xmm2[2],xmm3[3],xmm2[3]
 418 ; SSE2-NEXT:    psrad $16, %xmm3
 419 ; SSE2-NEXT:    punpckhwd {{.*#+}} xmm2 = xmm2[4,4,5,5,6,6,7,7]
 420 ; SSE2-NEXT:    psrad $16, %xmm2
 421 ; SSE2-NEXT:    movdqa %xmm2, dj+4112(%rax)
 422 ; SSE2-NEXT:    movdqa %xmm3, dj+4096(%rax)
 423 ; SSE2-NEXT:    addq $32, %rax
 424 ; SSE2-NEXT:    jne .LBB6_1
 425 ; SSE2-NEXT:  # %bb.2: # %for.end
 426 ; SSE2-NEXT:    retq
 427 ;
 428 ; SSE41-LABEL: example24:
 429 ; SSE41:       # %bb.0: # %vector.ph
 430 ; SSE41-NEXT:    movd %edi, %xmm0
 431 ; SSE41-NEXT:    pshuflw {{.*#+}} xmm0 = xmm0[0,0,0,0,4,5,6,7]
 432 ; SSE41-NEXT:    pshufd {{.*#+}} xmm1 = xmm0[0,0,0,0]
 433 ; SSE41-NEXT:    movd %esi, %xmm0
 434 ; SSE41-NEXT:    pshuflw {{.*#+}} xmm0 = xmm0[0,0,0,0,4,5,6,7]
 435 ; SSE41-NEXT:    pshufd {{.*#+}} xmm2 = xmm0[0,0,0,0]
 436 ; SSE41-NEXT:    movq $-4096, %rax # imm = 0xF000
 437 ; SSE41-NEXT:    .p2align 4
 438 ; SSE41-NEXT:  .LBB6_1: # %vector.body
 439 ; SSE41-NEXT:    # =>This Inner Loop Header: Depth=1
 440 ; SSE41-NEXT:    movaps da+4096(%rax), %xmm0
 441 ; SSE41-NEXT:    movaps da+4112(%rax), %xmm3
 442 ; SSE41-NEXT:    cmpltps db+4112(%rax), %xmm3
 443 ; SSE41-NEXT:    cmpltps db+4096(%rax), %xmm0
 444 ; SSE41-NEXT:    packssdw %xmm3, %xmm0
 445 ; SSE41-NEXT:    movdqa %xmm2, %xmm3
 446 ; SSE41-NEXT:    pblendvb %xmm0, %xmm1, %xmm3
 447 ; SSE41-NEXT:    pshufd {{.*#+}} xmm0 = xmm3[2,3,2,3]
 448 ; SSE41-NEXT:    pmovsxwd %xmm0, %xmm0
 449 ; SSE41-NEXT:    pmovsxwd %xmm3, %xmm3
 450 ; SSE41-NEXT:    movdqa %xmm3, dj+4096(%rax)
 451 ; SSE41-NEXT:    movdqa %xmm0, dj+4112(%rax)
 452 ; SSE41-NEXT:    addq $32, %rax
 453 ; SSE41-NEXT:    jne .LBB6_1
 454 ; SSE41-NEXT:  # %bb.2: # %for.end
 455 ; SSE41-NEXT:    retq
 456 ;
 457 ; AVX1-LABEL: example24:
 458 ; AVX1:       # %bb.0: # %vector.ph
 459 ; AVX1-NEXT:    vmovd %edi, %xmm0
 460 ; AVX1-NEXT:    vpshuflw {{.*#+}} xmm0 = xmm0[0,0,0,0,4,5,6,7]
 461 ; AVX1-NEXT:    vpshufd {{.*#+}} xmm2 = xmm0[0,0,0,0]
 462 ; AVX1-NEXT:    vmovd %esi, %xmm0
 463 ; AVX1-NEXT:    vpshuflw {{.*#+}} xmm0 = xmm0[0,0,0,0,4,5,6,7]
 464 ; AVX1-NEXT:    vpshufd {{.*#+}} xmm3 = xmm0[0,0,0,0]
 465 ; AVX1-NEXT:    movq $-4096, %rax # imm = 0xF000
 466 ; AVX1-NEXT:    vpmovsxwd %xmm2, %xmm0
 467 ; AVX1-NEXT:    vpmovsxwd %xmm3, %xmm1
 468 ; AVX1-NEXT:    vpshufd {{.*#+}} xmm2 = xmm2[2,3,2,3]
 469 ; AVX1-NEXT:    vpmovsxwd %xmm2, %xmm2
 470 ; AVX1-NEXT:    vpshufd {{.*#+}} xmm3 = xmm3[2,3,2,3]
 471 ; AVX1-NEXT:    vpmovsxwd %xmm3, %xmm3
 472 ; AVX1-NEXT:    .p2align 4
 473 ; AVX1-NEXT:  .LBB6_1: # %vector.body
 474 ; AVX1-NEXT:    # =>This Inner Loop Header: Depth=1
 475 ; AVX1-NEXT:    vmovups da+4096(%rax), %ymm4
 476 ; AVX1-NEXT:    vcmpltps db+4096(%rax), %ymm4, %ymm4
 477 ; AVX1-NEXT:    vblendvps %xmm4, %xmm0, %xmm1, %xmm5
 478 ; AVX1-NEXT:    vextractf128 $1, %ymm4, %xmm4
 479 ; AVX1-NEXT:    vblendvps %xmm4, %xmm2, %xmm3, %xmm4
 480 ; AVX1-NEXT:    vmovaps %xmm4, dj+4112(%rax)
 481 ; AVX1-NEXT:    vmovaps %xmm5, dj+4096(%rax)
 482 ; AVX1-NEXT:    addq $32, %rax
 483 ; AVX1-NEXT:    jne .LBB6_1
 484 ; AVX1-NEXT:  # %bb.2: # %for.end
 485 ; AVX1-NEXT:    vzeroupper
 486 ; AVX1-NEXT:    retq
 487 ;
 488 ; AVX2-LABEL: example24:
 489 ; AVX2:       # %bb.0: # %vector.ph
 490 ; AVX2-NEXT:    vmovd %edi, %xmm0
 491 ; AVX2-NEXT:    vpbroadcastw %xmm0, %xmm0
 492 ; AVX2-NEXT:    vmovd %esi, %xmm1
 493 ; AVX2-NEXT:    vpbroadcastw %xmm1, %xmm1
 494 ; AVX2-NEXT:    movq $-4096, %rax # imm = 0xF000
 495 ; AVX2-NEXT:    vpmovsxwd %xmm0, %ymm0
 496 ; AVX2-NEXT:    vpmovsxwd %xmm1, %ymm1
 497 ; AVX2-NEXT:    .p2align 4
 498 ; AVX2-NEXT:  .LBB6_1: # %vector.body
 499 ; AVX2-NEXT:    # =>This Inner Loop Header: Depth=1
 500 ; AVX2-NEXT:    vmovups da+4096(%rax), %ymm2
 501 ; AVX2-NEXT:    vcmpltps db+4096(%rax), %ymm2, %ymm2
 502 ; AVX2-NEXT:    vblendvps %ymm2, %ymm0, %ymm1, %ymm2
 503 ; AVX2-NEXT:    vmovups %ymm2, dj+4096(%rax)
 504 ; AVX2-NEXT:    addq $32, %rax
 505 ; AVX2-NEXT:    jne .LBB6_1
 506 ; AVX2-NEXT:  # %bb.2: # %for.end
 507 ; AVX2-NEXT:    vzeroupper
 508 ; AVX2-NEXT:    retq
 509 vector.ph:
 510   %0 = insertelement <8 x i16> undef, i16 %x, i32 0
 511   %broadcast11 = shufflevector <8 x i16> %0, <8 x i16> undef, <8 x i32> zeroinitializer
 512   %1 = insertelement <8 x i16> undef, i16 %y, i32 0
 513   %broadcast12 = shufflevector <8 x i16> %1, <8 x i16> undef, <8 x i32> zeroinitializer
 514   br label %vector.body
 515
 516 vector.body:
 517   %index = phi i64 [ 0, %vector.ph ], [ %index.next, %vector.body ]
 518   %2 = getelementptr inbounds [1024 x float], ptr @da, i64 0, i64 %index
 519   %3 = load <8 x float>, ptr %2, align 16
 520   %4 = getelementptr inbounds [1024 x float], ptr @db, i64 0, i64 %index
 521   %5 = load <8 x float>, ptr %4, align 16
 522   %6 = fcmp olt <8 x float> %3, %5
 523   %7 = select <8 x i1> %6, <8 x i16> %broadcast11, <8 x i16> %broadcast12
 524   %8 = sext <8 x i16> %7 to <8 x i32>
 525   %9 = getelementptr inbounds [1024 x i32], ptr @dj, i64 0, i64 %index
 526   store <8 x i32> %8, ptr %9, align 16
 527   %index.next = add i64 %index, 8
 528   %10 = icmp eq i64 %index.next, 1024
 529   br i1 %10, label %for.end, label %vector.body
 530
 531 for.end:
 532   ret void
 533 }
 534